24.02.1
|
#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 |
|
inline |
Returns consumer node.
Definition at line 102 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(), arm_compute::graph::detail::fuse_node_with_activation(), arm_compute::graph::detail::fuse_pad_with_convolution(), and Graph::remove_connection().
|
inline |
Returns sink node id.
Definition at line 86 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 118 of file Edge.h.
Referenced by Graph::add_connection(), and Graph::remove_connection().
|
inline |
Returns edge id.
Definition at line 70 of file Edge.h.
Referenced by Graph::add_connection().
|
inline |
Returns producer node.
Definition at line 94 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(), arm_compute::graph::detail::fuse_pad_with_convolution(), NodeFusionMutator::mutate(), and Graph::remove_connection().
|
inline |
Returns producer node id.
Definition at line 78 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 110 of file Edge.h.
Referenced by Graph::add_connection().
|
inline |
Returns the tensor associated with this edge.
Definition at line 126 of file Edge.h.
Referenced by arm_compute::graph::detail::fuse_pad_with_convolution(), INode::input(), InPlaceOperationMutator::mutate(), Graph::remove_connection(), INode::set_output_tensor(), and Edge::update_bound_tensor().
|
inline |
Returns the tensor id associated with this edge.
Definition at line 134 of file Edge.h.
References Tensor::id(), and arm_compute::graph::NullTensorID.
Referenced by INode::input_id().
|
inline |