21.02
|
#include <Edge.h>
Public Member Functions | |
Edge (EdgeID id, INode *producer, unsigned int producer_idx, INode *consumer, unsigned int consumer_idx, Tensor *tensor) | |
Default Constructor. More... | |
EdgeID | id () const |
Returns edge id. More... | |
NodeID | producer_id () const |
Returns producer node id. More... | |
NodeID | consumer_id () const |
Returns sink node id. More... | |
INode * | producer () const |
Returns producer node. More... | |
INode * | consumer () const |
Returns consumer node. More... | |
unsigned int | producer_idx () const |
Returns the index of the output that produces the result in the producer node. More... | |
unsigned int | consumer_idx () const |
Returns the index of the input that consumes the result in the consumer node. More... | |
Tensor * | tensor () const |
Returns the tensor associated with this edge. More... | |
TensorID | tensor_id () const |
Returns the tensor id associated with this edge. More... | |
void | update_bound_tensor (Tensor *tensor) |
Bind the edge to another tensor. More... | |
Friends | |
class | Graph |
|
inline |
Default Constructor.
[in] | id | Edge id |
[in] | producer | Producer node id |
[in] | producer_idx | Producer node output index |
[in] | consumer | Consumer node id |
[in] | consumer_idx | Consumer node input index |
[in] | tensor | Tensor associated with the edge |
Definition at line 51 of file Edge.h.
|
inline |
Returns consumer node.
Definition at line 92 of file Edge.h.
Referenced by arm_compute::graph::bfs(), arm_compute::graph::dfs(), arm_compute::graph::detail::fuse_convolution_with_batch_normalization(), arm_compute::graph::detail::fuse_depthwise_convolution_with_batch_normalization(), and arm_compute::graph::detail::fuse_node_with_activation().
|
inline |
Returns sink node id.
Definition at line 76 of file Edge.h.
References arm_compute::graph::EmptyNodeID, and INode::id().
Referenced by Graph::add_connection(), arm_compute::graph::bfs(), arm_compute::graph::dfs(), arm_compute::graph::detail::fuse_convolution_with_batch_normalization(), arm_compute::graph::detail::fuse_depthwise_convolution_with_batch_normalization(), and arm_compute::graph::detail::fuse_node_with_activation().
|
inline |
Returns the index of the input that consumes the result in the consumer node.
Definition at line 108 of file Edge.h.
Referenced by Graph::add_connection().
|
inline |
Returns edge id.
Definition at line 60 of file Edge.h.
Referenced by Graph::add_connection().
|
inline |
Returns producer node.
Definition at line 84 of file Edge.h.
Referenced by arm_compute::graph::detail::all_inputs_are_visited(), arm_compute::graph::detail::fuse_convolution_with_batch_normalization(), arm_compute::graph::detail::fuse_depthwise_convolution_with_batch_normalization(), arm_compute::graph::detail::fuse_node_with_activation(), and NodeFusionMutator::mutate().
|
inline |
Returns producer node id.
Definition at line 68 of file Edge.h.
References arm_compute::graph::EmptyNodeID, and INode::id().
Referenced by Graph::add_connection(), arm_compute::graph::detail::all_inputs_are_visited(), arm_compute::graph::detail::fuse_convolution_with_batch_normalization(), arm_compute::graph::detail::fuse_depthwise_convolution_with_batch_normalization(), and arm_compute::graph::detail::fuse_node_with_activation().
|
inline |
Returns the index of the output that produces the result in the producer node.
Definition at line 100 of file Edge.h.
Referenced by Graph::add_connection().
|
inline |
Returns the tensor associated with this edge.
Definition at line 116 of file Edge.h.
Referenced by INode::input(), InPlaceOperationMutator::mutate(), DepthConcatSubTensorMutator::mutate(), and INode::set_output_tensor().
|
inline |
Returns the tensor id associated with this edge.
Definition at line 124 of file Edge.h.
References Tensor::id(), and arm_compute::graph::NullTensorID.
Referenced by INode::input_id().
|
inline |
Bind the edge to another tensor.
[in] | tensor | Tensor to bind the edge to |
Definition at line 134 of file Edge.h.