21.02
|
#include <Tensor.h>
Public Member Functions | |
Tensor (TensorID id, TensorDescriptor desc) | |
Default constructor. More... | |
TensorID | id () const |
Tensor ID accessor. More... | |
TensorDescriptor & | desc () |
TensorInfo metadata accessor. More... | |
const TensorDescriptor & | desc () const |
TensorInfo metadata accessor. More... | |
void | set_handle (std::unique_ptr< ITensorHandle > backend_tensor) |
Sets the backend tensor. More... | |
ITensorHandle * | handle () |
Backend tensor handle accessor. More... | |
void | set_accessor (std::unique_ptr< ITensorAccessor > accessor) |
Sets the backend tensor accessor. More... | |
ITensorAccessor * | accessor () |
Backend tensor accessor. More... | |
std::unique_ptr< ITensorAccessor > | extract_accessor () |
Extracts accessor from the tensor. More... | |
bool | call_accessor () |
Calls accessor on tensor. More... | |
void | bind_edge (EdgeID eid) |
Binds the tensor with an edge. More... | |
void | unbind_edge (EdgeID eid) |
Unbinds an edge from a tensor. More... | |
std::set< EdgeID > | bound_edges () const |
Accessor the edges that are bound with the tensor. More... | |
Tensor | ( | TensorID | id, |
TensorDescriptor | desc | ||
) |
Default constructor.
Definition at line 30 of file Tensor.cpp.
ITensorAccessor * accessor | ( | ) |
void bind_edge | ( | EdgeID | eid | ) |
Binds the tensor with an edge.
[in] | eid | Edge ID that is bound to the tensor |
Definition at line 101 of file Tensor.cpp.
Referenced by INode::set_output_tensor().
std::set< EdgeID > bound_edges | ( | ) | const |
Accessor the edges that are bound with the tensor.
Definition at line 111 of file Tensor.cpp.
Referenced by arm_compute::graph::detail::allocate_all_input_tensors(), and arm_compute::graph::detail::allocate_all_output_tensors().
bool call_accessor | ( | ) |
Calls accessor on tensor.
Definition at line 75 of file Tensor.cpp.
Referenced by arm_compute::graph::detail::call_tensor_accessor().
TensorDescriptor & desc | ( | ) |
TensorInfo metadata accessor.
Definition at line 40 of file Tensor.cpp.
Referenced by PReluLayerNode::configure_output(), GenerateProposalsLayerNode::configure_output(), ROIAlignLayerNode::configure_output(), StackLayerNode::configure_output(), DeconvolutionLayerNode::configure_output(), ConcatenateLayerNode::configure_output(), EltwiseLayerNode::configure_output(), FullyConnectedLayerNode::configure_output(), DepthwiseConvolutionLayerNode::configure_output(), ConvolutionLayerNode::configure_output(), UnaryEltwiseLayerNode::configure_output(), FusedDepthwiseConvolutionBatchNormalizationNode::configure_output(), FusedConvolutionBatchNormalizationNode::configure_output(), arm_compute::graph::configure_tensor(), arm_compute::graph::backends::detail::create_concatenate_layer(), NEDeviceBackend::create_tensor(), GCDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), FlattenLayerNode::forward_descriptors(), NormalizePlanarYUVLayerNode::forward_descriptors(), PReluLayerNode::forward_descriptors(), ConstNode::forward_descriptors(), InputNode::forward_descriptors(), ReshapeLayerNode::forward_descriptors(), DequantizationLayerNode::forward_descriptors(), NormalizationLayerNode::forward_descriptors(), DetectionPostProcessLayerNode::forward_descriptors(), ChannelShuffleLayerNode::forward_descriptors(), SoftmaxLayerNode::forward_descriptors(), GenerateProposalsLayerNode::forward_descriptors(), BoundingBoxTransformLayerNode::forward_descriptors(), ResizeLayerNode::forward_descriptors(), ActivationLayerNode::forward_descriptors(), PermuteLayerNode::forward_descriptors(), QuantizationLayerNode::forward_descriptors(), DummyNode::forward_descriptors(), PadLayerNode::forward_descriptors(), ReductionLayerNode::forward_descriptors(), ROIAlignLayerNode::forward_descriptors(), DepthToSpaceLayerNode::forward_descriptors(), ReorgLayerNode::forward_descriptors(), PoolingLayerNode::forward_descriptors(), PriorBoxLayerNode::forward_descriptors(), DetectionOutputLayerNode::forward_descriptors(), StackLayerNode::forward_descriptors(), BatchNormalizationLayerNode::forward_descriptors(), DeconvolutionLayerNode::forward_descriptors(), ArgMinMaxLayerNode::forward_descriptors(), PrintLayerNode::forward_descriptors(), SliceLayerNode::forward_descriptors(), L2NormalizeLayerNode::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(), arm_compute::graph::backends::detail::get_backing_tensor(), arm_compute::graph::get_tensor_descriptor(), NodeFusionMutator::mutate(), DepthConcatSubTensorMutator::mutate(), SplitLayerSubTensorMutator::mutate(), DotGraphPrinter::print(), and SplitLayerNode::validate().
const TensorDescriptor & desc | ( | ) | const |
TensorInfo metadata accessor.
Definition at line 45 of file Tensor.cpp.
std::unique_ptr< ITensorAccessor > extract_accessor | ( | ) |
Extracts accessor from the tensor.
Definition at line 70 of file Tensor.cpp.
ITensorHandle * handle | ( | ) |
Backend tensor handle accessor.
Definition at line 55 of file Tensor.cpp.
Referenced by arm_compute::graph::detail::allocate_all_input_tensors(), arm_compute::graph::detail::allocate_all_output_tensors(), arm_compute::graph::configure_tensor(), arm_compute::graph::execute_task(), arm_compute::graph::backends::detail::get_backing_tensor(), arm_compute::graph::backends::detail::get_backing_tensor_info(), and SplitLayerSubTensorMutator::mutate().
TensorID id | ( | ) | const |
Tensor ID accessor.
Definition at line 35 of file Tensor.cpp.
Referenced by Edge::tensor_id().
void set_accessor | ( | std::unique_ptr< ITensorAccessor > | accessor | ) |
Sets the backend tensor accessor.
[in] | accessor | Accessor to set |
Definition at line 60 of file Tensor.cpp.
Referenced by GroupedConvolutionMutator::mutate(), and arm_compute::graph::set_tensor_accessor().
void set_handle | ( | std::unique_ptr< ITensorHandle > | backend_tensor | ) |
Sets the backend tensor.
[in] | backend_tensor | Backend tensor to set |
Definition at line 50 of file Tensor.cpp.
Referenced by arm_compute::graph::configure_tensor(), and SplitLayerSubTensorMutator::mutate().
void unbind_edge | ( | EdgeID | eid | ) |
Unbinds an edge from a tensor.
[in] | eid | Edge to unbind |
Definition at line 106 of file Tensor.cpp.
Referenced by INode::set_output_tensor().