24 #ifndef __ARM_COMPUTE_GRAPH_EDGE_H__ 25 #define __ARM_COMPUTE_GRAPH_EDGE_H__ 52 : _id(id), _producer(producer), _consumer(consumer), _producer_idx(producer_idx), _consumer_idx(consumer_idx), _tensor(tensor)
70 return (_producer ==
nullptr) ?
EmptyNodeID : _producer->
id();
78 return (_consumer ==
nullptr) ?
EmptyNodeID : _consumer->
id();
102 return _producer_idx;
110 return _consumer_idx;
136 _tensor = (tensor !=
nullptr) ? tensor : _tensor;
146 unsigned int _producer_idx;
147 unsigned int _consumer_idx;
void update_bound_tensor(Tensor *tensor)
Bind the edge to another tensor.
NodeID producer_id() const
Returns producer node id.
TensorID id() const
Tensor ID accessor.
Tensor * tensor() const
Returns the tensor associated with this edge.
This file contains all available output stages for GEMMLowp on OpenCL.
Edge(EdgeID id, INode *producer, unsigned int producer_idx, INode *consumer, unsigned int consumer_idx, Tensor *tensor)
Default Constructor.
NodeID consumer_id() const
Returns sink node id.
unsigned int producer_idx() const
Returns the index of the output that produces the result in the producer node.
unsigned int consumer_idx() const
Returns the index of the input that consumes the result in the consumer node.
NodeID id() const
Returns node's ID.
TensorID tensor_id() const
Returns the tensor id associated with this edge.
INode * producer() const
Returns producer node.
constexpr NodeID EmptyNodeID
Constant EdgeID specifying an equivalent of null edge.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
EdgeID id() const
Returns edge id.
INode * consumer() const
Returns consumer node.