24.02.1
|
Node interface. More...
#include <INode.h>
Public Member Functions | |
INode () | |
Constructor. More... | |
virtual | ~INode ()=default |
Destructor. More... | |
INode (const INode &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INode & | operator= (const INode &)=delete |
Prevent instances of this class from being copy assigned (As this class contains pointers) More... | |
INode (INode &&)=default | |
Allow instances of this class to be moved. More... | |
INode & | operator= (INode &&)=default |
Allow instances of this class to be move assigned. More... | |
virtual Status | validate () const |
Validate node. More... | |
virtual NodeType | type () const =0 |
Returns node's type. More... | |
virtual void | accept (INodeVisitor &v)=0 |
Accepts a node visitor. More... | |
virtual bool | forward_descriptors ()=0 |
Forwards descriptor information to outputs if possible. More... | |
virtual TensorDescriptor | configure_output (size_t idx) const =0 |
Calculates output configuration. More... | |
std::string | name () const |
Returns node's name. More... | |
NodeID | id () const |
Returns node's ID. More... | |
const Graph * | graph () const |
Returns node's Graph. More... | |
Graph * | graph () |
Returns node's Graph. More... | |
void | set_graph (Graph *g) |
Sets the graph that this node is registered to. More... | |
void | set_id (NodeID id) |
Sets the node id. More... | |
void | set_common_node_parameters (NodeParams common_params) |
Sets common node parameters. More... | |
void | set_requested_target (Target target) |
Sets target preference. More... | |
void | set_assigned_target (Target target) |
Sets the final execution target. More... | |
void | set_output_tensor (TensorID tid, size_t idx) |
Sets the output tensor of at a given index. More... | |
const std::vector< TensorID > & | inputs () const |
Returns inputs of the node. More... | |
const std::vector< TensorID > & | outputs () const |
Returns outputs of the node. More... | |
const std::vector< EdgeID > & | input_edges () const |
Returns input edge set. More... | |
const std::set< EdgeID > & | output_edges () const |
Returns output edge set. More... | |
TensorID | input_id (size_t idx) const |
Returns the tensor ID of a given input of the node. More... | |
TensorID | output_id (size_t idx) const |
Returns the tensor ID of a given output of the node. More... | |
Tensor * | input (size_t idx) const |
Returns the tensor of a given input of the node. More... | |
Tensor * | output (size_t idx) const |
Returns the tensor of a given output of the node. More... | |
EdgeID | input_edge_id (size_t idx) const |
Returns the edge ID of a given input of the node. More... | |
Edge * | input_edge (size_t idx) const |
Returns the edge of a given input of the node. More... | |
size_t | num_inputs () const |
Returns number of inputs of the node. More... | |
size_t | num_outputs () const |
Returns number of outputs of the node. More... | |
NodeParams | common_node_params () const |
Returns common node parameters. More... | |
Target | requested_target () const |
Returns requested target for this node. More... | |
Target | assigned_target () const |
Returns assigned target for this node. More... | |
Friends | |
class | Graph |
INode | ( | ) |
|
virtualdefault |
Destructor.
Prevent instances of this class from being copied (As this class contains pointers)
|
pure virtual |
Accepts a node visitor.
[in] | v | Visitor to accept |
Implemented in FusedConvolutionBatchNormalizationNode, FusedDepthwiseConvolutionBatchNormalizationNode, UnaryEltwiseLayerNode, ConvolutionLayerNode, DepthwiseConvolutionLayerNode, FullyConnectedLayerNode, StridedSliceLayerNode, ConcatenateLayerNode, EltwiseLayerNode, SplitLayerNode, PrintLayerNode, L2NormalizeLayerNode, SliceLayerNode, ArgMinMaxLayerNode, DeconvolutionLayerNode, BatchNormalizationLayerNode, DetectionOutputLayerNode, PriorBoxLayerNode, StackLayerNode, PoolingLayerNode, ReorgLayerNode, DepthToSpaceLayerNode, PadLayerNode, ReductionLayerNode, ROIAlignLayerNode, DummyNode, PermuteLayerNode, QuantizationLayerNode, ResizeLayerNode, ActivationLayerNode, BoundingBoxTransformLayerNode, GenerateProposalsLayerNode, ChannelShuffleLayerNode, DetectionPostProcessLayerNode, NormalizationLayerNode, SoftmaxLayerNode, DequantizationLayerNode, ConstNode, InputNode, ReshapeLayerNode, FlattenLayerNode, NormalizePlanarYUVLayerNode, OutputNode, and PReluLayerNode.
Target assigned_target | ( | ) | const |
Returns assigned target for this node.
Definition at line 198 of file INode.cpp.
Referenced by arm_compute::graph::detail::configure_all_nodes(), NEDeviceBackend::configure_node(), CLDeviceBackend::configure_node(), GroupedConvolutionMutator::mutate(), NEDeviceBackend::validate_node(), CLDeviceBackend::validate_node(), and arm_compute::graph::backends::detail::validate_node().
NodeParams common_node_params | ( | ) | const |
|
pure virtual |
Calculates output configuration.
[in] | idx | Output index to configure |
Implemented in FusedConvolutionBatchNormalizationNode, FusedDepthwiseConvolutionBatchNormalizationNode, UnaryEltwiseLayerNode, ConvolutionLayerNode, DepthwiseConvolutionLayerNode, FullyConnectedLayerNode, StridedSliceLayerNode, ConcatenateLayerNode, EltwiseLayerNode, SplitLayerNode, PrintLayerNode, L2NormalizeLayerNode, SliceLayerNode, ArgMinMaxLayerNode, DeconvolutionLayerNode, BatchNormalizationLayerNode, DetectionOutputLayerNode, PriorBoxLayerNode, StackLayerNode, PoolingLayerNode, ReorgLayerNode, DepthToSpaceLayerNode, PadLayerNode, ReductionLayerNode, ROIAlignLayerNode, DummyNode, PermuteLayerNode, QuantizationLayerNode, ResizeLayerNode, ActivationLayerNode, BoundingBoxTransformLayerNode, GenerateProposalsLayerNode, ChannelShuffleLayerNode, DetectionPostProcessLayerNode, NormalizationLayerNode, SoftmaxLayerNode, DequantizationLayerNode, ConstNode, InputNode, ReshapeLayerNode, FlattenLayerNode, NormalizePlanarYUVLayerNode, OutputNode, and PReluLayerNode.
|
pure virtual |
Forwards descriptor information to outputs if possible.
Implemented in FusedConvolutionBatchNormalizationNode, FusedDepthwiseConvolutionBatchNormalizationNode, UnaryEltwiseLayerNode, ConvolutionLayerNode, DepthwiseConvolutionLayerNode, FullyConnectedLayerNode, StridedSliceLayerNode, ConcatenateLayerNode, EltwiseLayerNode, SplitLayerNode, PrintLayerNode, L2NormalizeLayerNode, SliceLayerNode, ArgMinMaxLayerNode, DeconvolutionLayerNode, BatchNormalizationLayerNode, DetectionOutputLayerNode, PriorBoxLayerNode, StackLayerNode, PoolingLayerNode, ReorgLayerNode, DepthToSpaceLayerNode, PadLayerNode, ReductionLayerNode, ROIAlignLayerNode, DummyNode, PermuteLayerNode, QuantizationLayerNode, ResizeLayerNode, ActivationLayerNode, BoundingBoxTransformLayerNode, GenerateProposalsLayerNode, ChannelShuffleLayerNode, DetectionPostProcessLayerNode, NormalizationLayerNode, SoftmaxLayerNode, DequantizationLayerNode, ConstNode, InputNode, ReshapeLayerNode, FlattenLayerNode, NormalizePlanarYUVLayerNode, OutputNode, and PReluLayerNode.
Referenced by Graph::add_node().
Graph * graph | ( | ) | const |
Returns node's Graph.
Definition at line 112 of file INode.cpp.
Referenced by arm_compute::graph::detail::all_inputs_are_visited(), arm_compute::graph::get_driver_nodes(), and arm_compute::graph::get_driving_nodes().
NodeID id | ( | ) | const |
Returns node's ID.
Definition at line 102 of file INode.cpp.
Referenced by NEDeviceBackend::configure_node(), CLDeviceBackend::configure_node(), Edge::consumer_id(), arm_compute::graph::backends::detail::create_concatenate_layer(), arm_compute::graph::backends::detail::create_stack_layer(), DepthConcatSubTensorMutator::mutate(), GroupedConvolutionMutator::mutate(), SplitLayerSubTensorMutator::mutate(), NodeExecutionMethodMutator::mutate(), Edge::producer_id(), INode::set_id(), arm_compute::graph::detail::transfer_driving_nodes_and_remove_old_node(), arm_compute::graph::backends::detail::validate_arg_min_max_layer(), arm_compute::graph::backends::detail::validate_bounding_box_transform_layer(), arm_compute::graph::backends::detail::validate_channel_shuffle_layer(), arm_compute::graph::backends::detail::validate_convolution_layer(), arm_compute::graph::backends::detail::validate_depth_to_space_layer(), arm_compute::graph::backends::detail::validate_depthwise_convolution_layer(), arm_compute::graph::backends::detail::validate_dequantization_layer(), arm_compute::graph::backends::detail::validate_detection_output_layer(), arm_compute::graph::backends::detail::validate_detection_post_process_layer(), arm_compute::graph::backends::detail::validate_eltwise_Layer(), arm_compute::graph::backends::detail::validate_generate_proposals_layer(), arm_compute::graph::backends::detail::validate_l2_normalize_layer(), NEDeviceBackend::validate_node(), CLDeviceBackend::validate_node(), arm_compute::graph::backends::detail::validate_node(), arm_compute::graph::backends::detail::validate_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::validate_pad_layer(), arm_compute::graph::backends::detail::validate_permute_layer(), arm_compute::graph::backends::detail::validate_prelu_layer(), arm_compute::graph::backends::detail::validate_priorbox_layer(), arm_compute::graph::backends::detail::validate_quantization_layer(), arm_compute::graph::backends::detail::validate_reduction_operation_layer(), arm_compute::graph::backends::detail::validate_reorg_layer(), arm_compute::graph::backends::detail::validate_reshape_layer(), arm_compute::graph::backends::detail::validate_roi_align_layer(), arm_compute::graph::backends::detail::validate_slice_layer(), arm_compute::graph::backends::detail::validate_strided_slice_layer(), and arm_compute::graph::backends::detail::validate_unary_eltwise_layer().
Tensor * input | ( | size_t | idx | ) | const |
Returns the tensor of a given input of the node.
[in] | idx | Index of the node input |
Definition at line 150 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON, Graph::edge(), and Edge::tensor().
Referenced by arm_compute::graph::detail::allocate_all_input_tensors(), FlattenLayerNode::configure_output(), NormalizePlanarYUVLayerNode::configure_output(), PReluLayerNode::configure_output(), ReshapeLayerNode::configure_output(), DequantizationLayerNode::configure_output(), ChannelShuffleLayerNode::configure_output(), NormalizationLayerNode::configure_output(), SoftmaxLayerNode::configure_output(), ActivationLayerNode::configure_output(), BoundingBoxTransformLayerNode::configure_output(), GenerateProposalsLayerNode::configure_output(), DummyNode::configure_output(), PermuteLayerNode::configure_output(), QuantizationLayerNode::configure_output(), ResizeLayerNode::configure_output(), DepthToSpaceLayerNode::configure_output(), PadLayerNode::configure_output(), ReductionLayerNode::configure_output(), ROIAlignLayerNode::configure_output(), PoolingLayerNode::configure_output(), ReorgLayerNode::configure_output(), DetectionOutputLayerNode::configure_output(), PriorBoxLayerNode::configure_output(), BatchNormalizationLayerNode::configure_output(), DeconvolutionLayerNode::configure_output(), ArgMinMaxLayerNode::configure_output(), L2NormalizeLayerNode::configure_output(), SliceLayerNode::configure_output(), PrintLayerNode::configure_output(), SplitLayerNode::configure_output(), EltwiseLayerNode::configure_output(), StridedSliceLayerNode::configure_output(), FullyConnectedLayerNode::configure_output(), DepthwiseConvolutionLayerNode::configure_output(), ConvolutionLayerNode::configure_output(), UnaryEltwiseLayerNode::configure_output(), FusedDepthwiseConvolutionBatchNormalizationNode::configure_output(), FusedConvolutionBatchNormalizationNode::configure_output(), arm_compute::graph::backends::detail::create_activation_layer(), arm_compute::graph::backends::detail::create_arg_min_max_layer(), arm_compute::graph::backends::detail::create_batch_normalization_layer(), arm_compute::graph::backends::detail::create_bounding_box_transform_layer(), arm_compute::graph::backends::detail::create_channel_shuffle_layer(), arm_compute::graph::backends::detail::create_concatenate_layer(), arm_compute::graph::backends::detail::create_convolution_layer(), arm_compute::graph::backends::detail::create_deconvolution_layer(), arm_compute::graph::backends::detail::create_depth_to_space_layer(), arm_compute::graph::backends::detail::create_depthwise_convolution_layer(), arm_compute::graph::backends::detail::create_dequantization_layer(), arm_compute::graph::backends::detail::create_detection_output_layer(), arm_compute::graph::backends::detail::create_detection_output_layer< CPPDetectionOutputLayer, CLTargetInfo >(), arm_compute::graph::backends::detail::create_detection_post_process_layer(), arm_compute::graph::backends::detail::create_detection_post_process_layer< CPPDetectionPostProcessLayer, CLTargetInfo >(), arm_compute::graph::backends::detail::create_eltwise_layer(), arm_compute::graph::backends::detail::create_flatten_layer(), arm_compute::graph::backends::detail::create_fully_connected_layer(), arm_compute::graph::backends::detail::create_fused_convolution_batch_normalization_layer(), arm_compute::graph::backends::detail::create_fused_depthwise_convolution_batch_normalization_layer(), arm_compute::graph::backends::detail::create_generate_proposals_layer(), arm_compute::graph::backends::detail::create_l2_normalize_layer(), arm_compute::graph::backends::detail::create_normalization_layer(), arm_compute::graph::backends::detail::create_normalization_layer< NENormalizationLayer, NETargetInfo >(), arm_compute::graph::backends::detail::create_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::create_pad_layer(), arm_compute::graph::backends::detail::create_permute_layer(), arm_compute::graph::backends::detail::create_pooling_layer(), arm_compute::graph::backends::detail::create_prelu_layer(), arm_compute::graph::backends::detail::create_print_layer(), arm_compute::graph::backends::detail::create_priorbox_layer(), arm_compute::graph::backends::detail::create_quantization_layer(), arm_compute::graph::backends::detail::create_reduction_operation_layer(), arm_compute::graph::backends::detail::create_reorg_layer(), arm_compute::graph::backends::detail::create_reshape_layer(), arm_compute::graph::backends::detail::create_resize_layer(), arm_compute::graph::backends::detail::create_roi_align_layer(), arm_compute::graph::backends::detail::create_slice_layer(), arm_compute::graph::backends::detail::create_softmax_layer(), arm_compute::graph::backends::detail::create_stack_layer(), arm_compute::graph::backends::detail::create_strided_slice_layer(), arm_compute::graph::backends::detail::create_unary_eltwise_layer(), SplitLayerSubTensorMutator::mutate(), DepthConcatSubTensorMutator::mutate(), SplitLayerNode::validate(), arm_compute::graph::backends::detail::validate_arg_min_max_layer(), arm_compute::graph::backends::detail::validate_bounding_box_transform_layer(), arm_compute::graph::backends::detail::validate_channel_shuffle_layer(), arm_compute::graph::backends::detail::validate_convolution_layer(), arm_compute::graph::backends::detail::validate_depth_to_space_layer(), arm_compute::graph::backends::detail::validate_depthwise_convolution_layer(), arm_compute::graph::backends::detail::validate_dequantization_layer(), arm_compute::graph::backends::detail::validate_detection_output_layer(), arm_compute::graph::backends::detail::validate_detection_post_process_layer(), arm_compute::graph::backends::detail::validate_eltwise_Layer(), arm_compute::graph::backends::detail::validate_generate_proposals_layer(), arm_compute::graph::backends::detail::validate_l2_normalize_layer(), arm_compute::graph::backends::detail::validate_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::validate_pad_layer(), arm_compute::graph::backends::detail::validate_permute_layer(), arm_compute::graph::backends::detail::validate_prelu_layer(), arm_compute::graph::backends::detail::validate_priorbox_layer(), arm_compute::graph::backends::detail::validate_quantization_layer(), arm_compute::graph::backends::detail::validate_reduction_operation_layer(), arm_compute::graph::backends::detail::validate_reorg_layer(), arm_compute::graph::backends::detail::validate_reshape_layer(), arm_compute::graph::backends::detail::validate_roi_align_layer(), arm_compute::graph::backends::detail::validate_slice_layer(), arm_compute::graph::backends::detail::validate_strided_slice_layer(), and arm_compute::graph::backends::detail::validate_unary_eltwise_layer().
Edge * input_edge | ( | size_t | idx | ) | const |
Returns the edge of a given input of the node.
[in] | idx | Index of the node input |
Definition at line 171 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON, and Graph::edge().
Referenced by arm_compute::graph::detail::fuse_convolution_with_batch_normalization(), and arm_compute::graph::detail::fuse_depthwise_convolution_with_batch_normalization().
EdgeID input_edge_id | ( | size_t | idx | ) | const |
Returns the edge ID of a given input of the node.
[in] | idx | Index of the node input |
Definition at line 165 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON.
const std::vector< EdgeID > & input_edges | ( | ) | const |
Returns input edge set.
Definition at line 127 of file INode.cpp.
Referenced by arm_compute::graph::detail::all_inputs_are_visited(), arm_compute::graph::get_driver_nodes(), and DepthConcatSubTensorMutator::mutate().
TensorID input_id | ( | size_t | idx | ) | const |
Returns the tensor ID of a given input of the node.
[in] | idx | Index of the node input |
Definition at line 137 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON, Graph::edge(), arm_compute::graph::NullTensorID, and Edge::tensor_id().
Referenced by StackLayerNode::configure_output(), ConcatenateLayerNode::configure_output(), FlattenLayerNode::forward_descriptors(), NormalizePlanarYUVLayerNode::forward_descriptors(), PReluLayerNode::forward_descriptors(), ReshapeLayerNode::forward_descriptors(), DequantizationLayerNode::forward_descriptors(), SoftmaxLayerNode::forward_descriptors(), NormalizationLayerNode::forward_descriptors(), ChannelShuffleLayerNode::forward_descriptors(), DetectionPostProcessLayerNode::forward_descriptors(), GenerateProposalsLayerNode::forward_descriptors(), ActivationLayerNode::forward_descriptors(), BoundingBoxTransformLayerNode::forward_descriptors(), ResizeLayerNode::forward_descriptors(), PermuteLayerNode::forward_descriptors(), DummyNode::forward_descriptors(), QuantizationLayerNode::forward_descriptors(), ReductionLayerNode::forward_descriptors(), DepthToSpaceLayerNode::forward_descriptors(), ROIAlignLayerNode::forward_descriptors(), PadLayerNode::forward_descriptors(), ReorgLayerNode::forward_descriptors(), PoolingLayerNode::forward_descriptors(), PriorBoxLayerNode::forward_descriptors(), DetectionOutputLayerNode::forward_descriptors(), BatchNormalizationLayerNode::forward_descriptors(), DeconvolutionLayerNode::forward_descriptors(), ArgMinMaxLayerNode::forward_descriptors(), L2NormalizeLayerNode::forward_descriptors(), SliceLayerNode::forward_descriptors(), PrintLayerNode::forward_descriptors(), SplitLayerNode::forward_descriptors(), EltwiseLayerNode::forward_descriptors(), StridedSliceLayerNode::forward_descriptors(), FullyConnectedLayerNode::forward_descriptors(), DepthwiseConvolutionLayerNode::forward_descriptors(), ConvolutionLayerNode::forward_descriptors(), UnaryEltwiseLayerNode::forward_descriptors(), FusedDepthwiseConvolutionBatchNormalizationNode::forward_descriptors(), and FusedConvolutionBatchNormalizationNode::forward_descriptors().
const std::vector<TensorID>& inputs | ( | ) | const |
Returns inputs of the node.
std::string name | ( | ) | const |
Returns node's name.
Definition at line 107 of file INode.cpp.
References NodeParams::name.
Referenced by arm_compute::graph::backends::detail::create_activation_layer(), arm_compute::graph::backends::detail::create_arg_min_max_layer(), arm_compute::graph::backends::detail::create_batch_normalization_layer(), arm_compute::graph::backends::detail::create_bounding_box_transform_layer(), arm_compute::graph::backends::detail::create_channel_shuffle_layer(), arm_compute::graph::backends::detail::create_concatenate_layer(), arm_compute::graph::backends::detail::create_convolution_layer(), arm_compute::graph::backends::detail::create_deconvolution_layer(), arm_compute::graph::backends::detail::create_depth_to_space_layer(), arm_compute::graph::backends::detail::create_depthwise_convolution_layer(), arm_compute::graph::backends::detail::create_dequantization_layer(), arm_compute::graph::backends::detail::create_detection_output_layer(), arm_compute::graph::backends::detail::create_detection_output_layer< CPPDetectionOutputLayer, CLTargetInfo >(), arm_compute::graph::backends::detail::create_detection_post_process_layer(), arm_compute::graph::backends::detail::create_detection_post_process_layer< CPPDetectionPostProcessLayer, CLTargetInfo >(), arm_compute::graph::backends::detail::create_eltwise_layer(), arm_compute::graph::backends::detail::create_flatten_layer(), arm_compute::graph::backends::detail::create_fully_connected_layer(), arm_compute::graph::backends::detail::create_fused_convolution_batch_normalization_layer(), arm_compute::graph::backends::detail::create_fused_depthwise_convolution_batch_normalization_layer(), arm_compute::graph::backends::detail::create_l2_normalize_layer(), arm_compute::graph::backends::detail::create_normalization_layer(), arm_compute::graph::backends::detail::create_normalization_layer< NENormalizationLayer, NETargetInfo >(), arm_compute::graph::backends::detail::create_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::create_pad_layer(), arm_compute::graph::backends::detail::create_permute_layer(), arm_compute::graph::backends::detail::create_pooling_layer(), arm_compute::graph::backends::detail::create_prelu_layer(), arm_compute::graph::backends::detail::create_print_layer(), arm_compute::graph::backends::detail::create_priorbox_layer(), arm_compute::graph::backends::detail::create_quantization_layer(), arm_compute::graph::backends::detail::create_reduction_operation_layer(), arm_compute::graph::backends::detail::create_reorg_layer(), arm_compute::graph::backends::detail::create_reshape_layer(), arm_compute::graph::backends::detail::create_resize_layer(), arm_compute::graph::backends::detail::create_roi_align_layer(), arm_compute::graph::backends::detail::create_slice_layer(), arm_compute::graph::backends::detail::create_softmax_layer(), arm_compute::graph::backends::detail::create_stack_layer(), arm_compute::graph::backends::detail::create_strided_slice_layer(), arm_compute::graph::backends::detail::create_unary_eltwise_layer(), arm_compute::graph::detail::fuse_convolution_with_batch_normalization(), arm_compute::graph::detail::fuse_depthwise_convolution_with_batch_normalization(), DepthConcatSubTensorMutator::mutate(), SplitLayerSubTensorMutator::mutate(), NodeExecutionMethodMutator::mutate(), arm_compute::graph::backends::detail::validate_arg_min_max_layer(), arm_compute::graph::backends::detail::validate_bounding_box_transform_layer(), arm_compute::graph::backends::detail::validate_channel_shuffle_layer(), arm_compute::graph::backends::detail::validate_convolution_layer(), arm_compute::graph::backends::detail::validate_depth_to_space_layer(), arm_compute::graph::backends::detail::validate_depthwise_convolution_layer(), arm_compute::graph::backends::detail::validate_dequantization_layer(), arm_compute::graph::backends::detail::validate_detection_output_layer(), arm_compute::graph::backends::detail::validate_detection_post_process_layer(), arm_compute::graph::backends::detail::validate_eltwise_Layer(), arm_compute::graph::backends::detail::validate_generate_proposals_layer(), arm_compute::graph::backends::detail::validate_l2_normalize_layer(), arm_compute::graph::backends::detail::validate_node(), arm_compute::graph::backends::detail::validate_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::validate_pad_layer(), arm_compute::graph::backends::detail::validate_permute_layer(), arm_compute::graph::backends::detail::validate_prelu_layer(), arm_compute::graph::backends::detail::validate_priorbox_layer(), arm_compute::graph::backends::detail::validate_quantization_layer(), arm_compute::graph::backends::detail::validate_reduction_operation_layer(), arm_compute::graph::backends::detail::validate_reorg_layer(), arm_compute::graph::backends::detail::validate_reshape_layer(), arm_compute::graph::backends::detail::validate_roi_align_layer(), arm_compute::graph::backends::detail::validate_slice_layer(), arm_compute::graph::backends::detail::validate_strided_slice_layer(), and arm_compute::graph::backends::detail::validate_unary_eltwise_layer().
size_t num_inputs | ( | ) | const |
Returns number of inputs of the node.
Definition at line 178 of file INode.cpp.
Referenced by arm_compute::graph::detail::allocate_all_input_tensors(), arm_compute::graph::backends::detail::create_concatenate_layer(), arm_compute::graph::backends::detail::create_stack_layer(), arm_compute::graph::backends::detail::validate_arg_min_max_layer(), arm_compute::graph::backends::detail::validate_bounding_box_transform_layer(), arm_compute::graph::backends::detail::validate_channel_shuffle_layer(), arm_compute::graph::backends::detail::validate_convolution_layer(), arm_compute::graph::backends::detail::validate_depth_to_space_layer(), arm_compute::graph::backends::detail::validate_depthwise_convolution_layer(), arm_compute::graph::backends::detail::validate_dequantization_layer(), arm_compute::graph::backends::detail::validate_detection_output_layer(), arm_compute::graph::backends::detail::validate_detection_post_process_layer(), arm_compute::graph::backends::detail::validate_eltwise_Layer(), arm_compute::graph::backends::detail::validate_generate_proposals_layer(), arm_compute::graph::backends::detail::validate_l2_normalize_layer(), arm_compute::graph::backends::detail::validate_node(), arm_compute::graph::backends::detail::validate_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::validate_pad_layer(), arm_compute::graph::backends::detail::validate_permute_layer(), arm_compute::graph::backends::detail::validate_prelu_layer(), arm_compute::graph::backends::detail::validate_priorbox_layer(), arm_compute::graph::backends::detail::validate_quantization_layer(), arm_compute::graph::backends::detail::validate_reduction_operation_layer(), arm_compute::graph::backends::detail::validate_reorg_layer(), arm_compute::graph::backends::detail::validate_reshape_layer(), arm_compute::graph::backends::detail::validate_roi_align_layer(), arm_compute::graph::backends::detail::validate_slice_layer(), arm_compute::graph::backends::detail::validate_strided_slice_layer(), and arm_compute::graph::backends::detail::validate_unary_eltwise_layer().
size_t num_outputs | ( | ) | const |
Returns number of outputs of the node.
Definition at line 183 of file INode.cpp.
Referenced by arm_compute::graph::detail::allocate_all_output_tensors(), FullyConnectedLayerNode::compute_output_descriptor(), FullyConnectedLayerNode::compute_weights_descriptor(), arm_compute::graph::backends::detail::create_concatenate_layer(), arm_compute::graph::backends::detail::create_stack_layer(), arm_compute::graph::backends::detail::validate_arg_min_max_layer(), arm_compute::graph::backends::detail::validate_bounding_box_transform_layer(), arm_compute::graph::backends::detail::validate_channel_shuffle_layer(), arm_compute::graph::backends::detail::validate_convolution_layer(), arm_compute::graph::backends::detail::validate_depth_to_space_layer(), arm_compute::graph::backends::detail::validate_depthwise_convolution_layer(), arm_compute::graph::backends::detail::validate_dequantization_layer(), arm_compute::graph::backends::detail::validate_detection_output_layer(), arm_compute::graph::backends::detail::validate_detection_post_process_layer(), arm_compute::graph::backends::detail::validate_eltwise_Layer(), arm_compute::graph::backends::detail::validate_generate_proposals_layer(), arm_compute::graph::backends::detail::validate_l2_normalize_layer(), arm_compute::graph::backends::detail::validate_node(), arm_compute::graph::backends::detail::validate_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::validate_pad_layer(), arm_compute::graph::backends::detail::validate_permute_layer(), arm_compute::graph::backends::detail::validate_prelu_layer(), arm_compute::graph::backends::detail::validate_priorbox_layer(), arm_compute::graph::backends::detail::validate_quantization_layer(), arm_compute::graph::backends::detail::validate_reduction_operation_layer(), arm_compute::graph::backends::detail::validate_reorg_layer(), arm_compute::graph::backends::detail::validate_reshape_layer(), arm_compute::graph::backends::detail::validate_roi_align_layer(), arm_compute::graph::backends::detail::validate_slice_layer(), arm_compute::graph::backends::detail::validate_strided_slice_layer(), and arm_compute::graph::backends::detail::validate_unary_eltwise_layer().
Prevent instances of this class from being copy assigned (As this class contains pointers)
Tensor * output | ( | size_t | idx | ) | const |
Returns the tensor of a given output of the node.
[in] | idx | Index of the node output |
Definition at line 158 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON, and Graph::tensor().
Referenced by arm_compute::graph::detail::allocate_all_output_tensors(), arm_compute::graph::backends::detail::create_activation_layer(), arm_compute::graph::backends::detail::create_arg_min_max_layer(), arm_compute::graph::backends::detail::create_batch_normalization_layer(), arm_compute::graph::backends::detail::create_bounding_box_transform_layer(), arm_compute::graph::backends::detail::create_channel_shuffle_layer(), arm_compute::graph::backends::detail::create_concatenate_layer(), arm_compute::graph::backends::detail::create_convolution_layer(), arm_compute::graph::backends::detail::create_deconvolution_layer(), arm_compute::graph::backends::detail::create_depth_to_space_layer(), arm_compute::graph::backends::detail::create_depthwise_convolution_layer(), arm_compute::graph::backends::detail::create_dequantization_layer(), arm_compute::graph::backends::detail::create_detection_output_layer(), arm_compute::graph::backends::detail::create_detection_output_layer< CPPDetectionOutputLayer, CLTargetInfo >(), arm_compute::graph::backends::detail::create_detection_post_process_layer(), arm_compute::graph::backends::detail::create_detection_post_process_layer< CPPDetectionPostProcessLayer, CLTargetInfo >(), arm_compute::graph::backends::detail::create_eltwise_layer(), arm_compute::graph::backends::detail::create_flatten_layer(), arm_compute::graph::backends::detail::create_fully_connected_layer(), arm_compute::graph::backends::detail::create_fused_convolution_batch_normalization_layer(), arm_compute::graph::backends::detail::create_fused_depthwise_convolution_batch_normalization_layer(), arm_compute::graph::backends::detail::create_generate_proposals_layer(), arm_compute::graph::backends::detail::create_l2_normalize_layer(), arm_compute::graph::backends::detail::create_normalization_layer(), arm_compute::graph::backends::detail::create_normalization_layer< NENormalizationLayer, NETargetInfo >(), arm_compute::graph::backends::detail::create_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::create_pad_layer(), arm_compute::graph::backends::detail::create_permute_layer(), arm_compute::graph::backends::detail::create_pooling_layer(), arm_compute::graph::backends::detail::create_prelu_layer(), arm_compute::graph::backends::detail::create_priorbox_layer(), arm_compute::graph::backends::detail::create_quantization_layer(), arm_compute::graph::backends::detail::create_reduction_operation_layer(), arm_compute::graph::backends::detail::create_reorg_layer(), arm_compute::graph::backends::detail::create_reshape_layer(), arm_compute::graph::backends::detail::create_resize_layer(), arm_compute::graph::backends::detail::create_roi_align_layer(), arm_compute::graph::backends::detail::create_slice_layer(), arm_compute::graph::backends::detail::create_softmax_layer(), arm_compute::graph::backends::detail::create_stack_layer(), arm_compute::graph::backends::detail::create_strided_slice_layer(), arm_compute::graph::backends::detail::create_unary_eltwise_layer(), FlattenLayerNode::forward_descriptors(), NormalizePlanarYUVLayerNode::forward_descriptors(), PReluLayerNode::forward_descriptors(), ConstNode::forward_descriptors(), InputNode::forward_descriptors(), ReshapeLayerNode::forward_descriptors(), DequantizationLayerNode::forward_descriptors(), ChannelShuffleLayerNode::forward_descriptors(), DetectionPostProcessLayerNode::forward_descriptors(), NormalizationLayerNode::forward_descriptors(), SoftmaxLayerNode::forward_descriptors(), ActivationLayerNode::forward_descriptors(), BoundingBoxTransformLayerNode::forward_descriptors(), GenerateProposalsLayerNode::forward_descriptors(), DummyNode::forward_descriptors(), ResizeLayerNode::forward_descriptors(), PermuteLayerNode::forward_descriptors(), QuantizationLayerNode::forward_descriptors(), DepthToSpaceLayerNode::forward_descriptors(), PadLayerNode::forward_descriptors(), ReductionLayerNode::forward_descriptors(), ROIAlignLayerNode::forward_descriptors(), PoolingLayerNode::forward_descriptors(), ReorgLayerNode::forward_descriptors(), DetectionOutputLayerNode::forward_descriptors(), PriorBoxLayerNode::forward_descriptors(), StackLayerNode::forward_descriptors(), BatchNormalizationLayerNode::forward_descriptors(), DeconvolutionLayerNode::forward_descriptors(), ArgMinMaxLayerNode::forward_descriptors(), L2NormalizeLayerNode::forward_descriptors(), SliceLayerNode::forward_descriptors(), PrintLayerNode::forward_descriptors(), SplitLayerNode::forward_descriptors(), ConcatenateLayerNode::forward_descriptors(), EltwiseLayerNode::forward_descriptors(), StridedSliceLayerNode::forward_descriptors(), FullyConnectedLayerNode::forward_descriptors(), DepthwiseConvolutionLayerNode::forward_descriptors(), ConvolutionLayerNode::forward_descriptors(), UnaryEltwiseLayerNode::forward_descriptors(), FusedDepthwiseConvolutionBatchNormalizationNode::forward_descriptors(), FusedConvolutionBatchNormalizationNode::forward_descriptors(), NodeFusionMutator::mutate(), DepthConcatSubTensorMutator::mutate(), SplitLayerSubTensorMutator::mutate(), GroupedConvolutionMutator::mutate(), arm_compute::graph::detail::transfer_driving_nodes_and_remove_old_node(), arm_compute::graph::backends::detail::validate_arg_min_max_layer(), arm_compute::graph::backends::detail::validate_bounding_box_transform_layer(), arm_compute::graph::backends::detail::validate_channel_shuffle_layer(), arm_compute::graph::backends::detail::validate_convolution_layer(), arm_compute::graph::backends::detail::validate_depth_to_space_layer(), arm_compute::graph::backends::detail::validate_depthwise_convolution_layer(), arm_compute::graph::backends::detail::validate_dequantization_layer(), arm_compute::graph::backends::detail::validate_detection_output_layer(), arm_compute::graph::backends::detail::validate_detection_post_process_layer(), arm_compute::graph::backends::detail::validate_eltwise_Layer(), arm_compute::graph::backends::detail::validate_generate_proposals_layer(), arm_compute::graph::backends::detail::validate_l2_normalize_layer(), arm_compute::graph::backends::detail::validate_normalize_planar_yuv_layer(), arm_compute::graph::backends::detail::validate_pad_layer(), arm_compute::graph::backends::detail::validate_permute_layer(), arm_compute::graph::backends::detail::validate_prelu_layer(), arm_compute::graph::backends::detail::validate_priorbox_layer(), arm_compute::graph::backends::detail::validate_quantization_layer(), arm_compute::graph::backends::detail::validate_reduction_operation_layer(), arm_compute::graph::backends::detail::validate_reorg_layer(), arm_compute::graph::backends::detail::validate_reshape_layer(), arm_compute::graph::backends::detail::validate_roi_align_layer(), arm_compute::graph::backends::detail::validate_slice_layer(), arm_compute::graph::backends::detail::validate_strided_slice_layer(), and arm_compute::graph::backends::detail::validate_unary_eltwise_layer().
const std::set< EdgeID > & output_edges | ( | ) | const |
Returns output edge set.
Definition at line 132 of file INode.cpp.
Referenced by arm_compute::graph::bfs(), arm_compute::graph::dfs(), arm_compute::graph::detail::fuse_layer(), arm_compute::graph::get_driving_nodes(), and Graph::remove_node().
TensorID output_id | ( | size_t | idx | ) | const |
Returns the tensor ID of a given output of the node.
[in] | idx | Index of the node output |
Definition at line 144 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON.
Referenced by FlattenLayerNode::forward_descriptors(), NormalizePlanarYUVLayerNode::forward_descriptors(), PReluLayerNode::forward_descriptors(), ConstNode::forward_descriptors(), ReshapeLayerNode::forward_descriptors(), InputNode::forward_descriptors(), DequantizationLayerNode::forward_descriptors(), SoftmaxLayerNode::forward_descriptors(), ChannelShuffleLayerNode::forward_descriptors(), DetectionPostProcessLayerNode::forward_descriptors(), NormalizationLayerNode::forward_descriptors(), GenerateProposalsLayerNode::forward_descriptors(), ActivationLayerNode::forward_descriptors(), BoundingBoxTransformLayerNode::forward_descriptors(), QuantizationLayerNode::forward_descriptors(), ResizeLayerNode::forward_descriptors(), PermuteLayerNode::forward_descriptors(), DummyNode::forward_descriptors(), DepthToSpaceLayerNode::forward_descriptors(), ReductionLayerNode::forward_descriptors(), ROIAlignLayerNode::forward_descriptors(), PadLayerNode::forward_descriptors(), ReorgLayerNode::forward_descriptors(), PoolingLayerNode::forward_descriptors(), DetectionOutputLayerNode::forward_descriptors(), PriorBoxLayerNode::forward_descriptors(), BatchNormalizationLayerNode::forward_descriptors(), DeconvolutionLayerNode::forward_descriptors(), ArgMinMaxLayerNode::forward_descriptors(), SliceLayerNode::forward_descriptors(), L2NormalizeLayerNode::forward_descriptors(), PrintLayerNode::forward_descriptors(), SplitLayerNode::forward_descriptors(), EltwiseLayerNode::forward_descriptors(), StridedSliceLayerNode::forward_descriptors(), FullyConnectedLayerNode::forward_descriptors(), DepthwiseConvolutionLayerNode::forward_descriptors(), ConvolutionLayerNode::forward_descriptors(), UnaryEltwiseLayerNode::forward_descriptors(), FusedDepthwiseConvolutionBatchNormalizationNode::forward_descriptors(), and FusedConvolutionBatchNormalizationNode::forward_descriptors().
const std::vector< TensorID > & outputs | ( | ) | const |
Returns outputs of the node.
Definition at line 122 of file INode.cpp.
Referenced by GraphBuilder::add_batch_normalization_node(), GraphBuilder::add_convolution_node(), GraphBuilder::add_deconvolution_node(), GraphBuilder::add_depthwise_convolution_node(), GraphBuilder::add_detection_post_process_node(), GraphBuilder::add_fully_connected_layer(), GraphBuilder::add_normalize_planar_yuv_node(), GraphBuilder::add_scale_layer(), GraphBuilder::add_yolo_node(), and SplitLayerSubTensorMutator::mutate().
Target requested_target | ( | ) | const |
Returns requested target for this node.
Definition at line 193 of file INode.cpp.
References NodeParams::target.
void set_assigned_target | ( | Target | target | ) |
Sets the final execution target.
[in] | target | Final execution target |
void set_common_node_parameters | ( | NodeParams | common_params | ) |
Sets common node parameters.
[in] | common_params | Common node parameters to set |
Definition at line 61 of file INode.cpp.
Referenced by GraphEdsr::setup().
void set_graph | ( | Graph * | g | ) |
Sets the graph that this node is registered to.
[in] | g | Back reference to graph |
Definition at line 50 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON.
Referenced by Graph::add_node().
void set_id | ( | NodeID | id | ) |
Sets the node id.
[in] | id | Node id |
Definition at line 56 of file INode.cpp.
References INode::id().
Referenced by Graph::add_node().
void set_output_tensor | ( | TensorID | tid, |
size_t | idx | ||
) |
Sets the output tensor of at a given index.
[in] | tid | Tensor ID |
[in] | idx | Output index |
Definition at line 76 of file INode.cpp.
References ARM_COMPUTE_ERROR_ON, Tensor::bind_edge(), Graph::edge(), arm_compute::graph::NullTensorID, Edge::tensor(), Graph::tensor(), and Tensor::unbind_edge().
void set_requested_target | ( | Target | target | ) |
Sets target preference.
[in] | target | Target preference |
Definition at line 66 of file INode.cpp.
References NodeParams::target.
|
pure virtual |
Returns node's type.
Implemented in FusedConvolutionBatchNormalizationNode, FusedDepthwiseConvolutionBatchNormalizationNode, UnaryEltwiseLayerNode, ConvolutionLayerNode, DepthwiseConvolutionLayerNode, FullyConnectedLayerNode, StridedSliceLayerNode, ConcatenateLayerNode, EltwiseLayerNode, SplitLayerNode, PrintLayerNode, L2NormalizeLayerNode, SliceLayerNode, ArgMinMaxLayerNode, DeconvolutionLayerNode, BatchNormalizationLayerNode, DetectionOutputLayerNode, PriorBoxLayerNode, StackLayerNode, PoolingLayerNode, ReorgLayerNode, DepthToSpaceLayerNode, PadLayerNode, ReductionLayerNode, ROIAlignLayerNode, DummyNode, PermuteLayerNode, QuantizationLayerNode, ResizeLayerNode, ActivationLayerNode, BoundingBoxTransformLayerNode, GenerateProposalsLayerNode, ChannelShuffleLayerNode, DetectionPostProcessLayerNode, NormalizationLayerNode, SoftmaxLayerNode, DequantizationLayerNode, ConstNode, InputNode, ReshapeLayerNode, FlattenLayerNode, NormalizePlanarYUVLayerNode, OutputNode, and PReluLayerNode.
Referenced by Graph::add_node(), CLFunctionFactory::create(), NEFunctionFactory::create(), arm_compute::graph::execute_task(), arm_compute::graph::is_utility_node(), DepthConcatSubTensorMutator::mutate(), GroupedConvolutionMutator::mutate(), SplitLayerSubTensorMutator::mutate(), Graph::remove_node(), NENodeValidator::validate(), CLNodeValidator::validate(), and arm_compute::graph::backends::detail::validate_node().
|
virtual |