21.08
|
Tensor metadata class. More...
#include <TensorDescriptor.h>
Public Member Functions | |
TensorDescriptor ()=default | |
Default Constructor. More... | |
TensorDescriptor (TensorShape tensor_shape, DataType tensor_data_type, QuantizationInfo tensor_quant_info=QuantizationInfo(), DataLayout tensor_data_layout=DataLayout::NCHW, Target tensor_target=Target::UNSPECIFIED) | |
Constructor. More... | |
TensorDescriptor & | set_shape (TensorShape &tensor_shape) |
Sets tensor descriptor shape. More... | |
TensorDescriptor & | set_data_type (DataType tensor_data_type) |
Sets tensor descriptor data type. More... | |
TensorDescriptor & | set_layout (DataLayout data_layout) |
Sets tensor descriptor data layout. More... | |
TensorDescriptor & | set_quantization_info (QuantizationInfo tensor_quant_info) |
Sets tensor descriptor quantization info. More... | |
std::unique_ptr< TensorDescriptor > | clone () const override |
Provide a clone of the current object of class T. More... | |
![]() | |
virtual | ~ICloneable ()=default |
Default virtual desctructor. More... | |
Data Fields | |
TensorShape | shape {} |
Tensor shape. More... | |
DataType | data_type { DataType::UNKNOWN } |
Data type. More... | |
DataLayout | layout { DataLayout::NCHW } |
Data layout. More... | |
QuantizationInfo | quant_info {} |
Quantization info. More... | |
Target | target { Target::UNSPECIFIED } |
Target. More... | |
Tensor metadata class.
Definition at line 38 of file TensorDescriptor.h.
|
default |
Default Constructor.
|
inline |
Constructor.
[in] | tensor_shape | Tensor shape |
[in] | tensor_data_type | Tensor data type |
[in] | tensor_quant_info | Tensor quantization info |
[in] | tensor_data_layout | Tensor data layout |
[in] | tensor_target | Target to allocate the tensor for |
Definition at line 50 of file TensorDescriptor.h.
|
inlineoverridevirtual |
Provide a clone of the current object of class T.
Implements ICloneable< TensorDescriptor >.
Definition at line 104 of file TensorDescriptor.h.
|
inline |
Sets tensor descriptor data type.
[in] | tensor_data_type | Data type |
Definition at line 75 of file TensorDescriptor.h.
References TensorDescriptor::data_type.
|
inline |
Sets tensor descriptor data layout.
[in] | data_layout | Data layout |
Definition at line 86 of file TensorDescriptor.h.
References arm_compute::test::validation::data_layout, and TensorDescriptor::layout.
|
inline |
Sets tensor descriptor quantization info.
[in] | tensor_quant_info | Quantization information |
Definition at line 97 of file TensorDescriptor.h.
References TensorDescriptor::quant_info.
Referenced by DeconvolutionLayerNode::configure_output().
|
inline |
Sets tensor descriptor shape.
[in] | tensor_shape | Tensor shape to set |
Definition at line 64 of file TensorDescriptor.h.
References TensorDescriptor::shape.
Referenced by ReductionLayerNode::configure_output(), and ArgMinMaxLayerNode::configure_output().
DataType data_type { DataType::UNKNOWN } |
Data type.
Definition at line 110 of file TensorDescriptor.h.
Referenced by GraphBuilder::add_convolution_node(), GraphBuilder::add_deconvolution_node(), GraphBuilder::add_depthwise_convolution_node(), GraphBuilder::add_fully_connected_layer(), DequantizationLayerNode::configure_output(), SoftmaxLayerNode::configure_output(), GenerateProposalsLayerNode::configure_output(), QuantizationLayerNode::configure_output(), ArgMinMaxLayerNode::configure_output(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), DotGraphPrinter::print(), and TensorDescriptor::set_data_type().
DataLayout layout { DataLayout::NCHW } |
Data layout.
Definition at line 111 of file TensorDescriptor.h.
Referenced by GraphBuilder::add_convolution_node(), GraphBuilder::add_deconvolution_node(), GraphBuilder::add_depthwise_convolution_node(), GraphBuilder::add_scale_layer(), GraphBuilder::add_yolo_node(), DepthToSpaceLayerNode::compute_output_descriptor(), PoolingLayerNode::compute_output_descriptor(), ReorgLayerNode::compute_output_descriptor(), DeconvolutionLayerNode::compute_output_descriptor(), FusedConvolutionBatchNormalizationNode::compute_output_descriptor(), DepthwiseConvolutionLayerNode::compute_output_descriptor(), FusedDepthwiseConvolutionBatchNormalizationNode::compute_output_descriptor(), ConvolutionLayerNode::compute_output_descriptor(), PermuteLayerNode::configure_output(), ROIAlignLayerNode::configure_output(), arm_compute::graph::backends::detail::create_concatenate_layer(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), arm_compute::graph::get_dimension_size(), DotGraphPrinter::print(), and TensorDescriptor::set_layout().
QuantizationInfo quant_info {} |
Quantization info.
Definition at line 112 of file TensorDescriptor.h.
Referenced by GraphBuilder::add_convolution_node(), GraphBuilder::add_depthwise_convolution_node(), GraphBuilder::add_detection_post_process_node(), FullyConnectedLayerNode::compute_output_descriptor(), FullyConnectedLayerNode::compute_weights_descriptor(), SoftmaxLayerNode::configure_output(), ActivationLayerNode::configure_output(), QuantizationLayerNode::configure_output(), ArgMinMaxLayerNode::configure_output(), DepthwiseConvolutionLayerNode::configure_output(), ConvolutionLayerNode::configure_output(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), NodeFusionMutator::mutate(), DepthConcatSubTensorMutator::mutate(), and TensorDescriptor::set_quantization_info().
TensorShape shape {} |
Tensor shape.
Definition at line 109 of file TensorDescriptor.h.
Referenced by GraphBuilder::add_batch_normalization_node(), GraphBuilder::add_convolution_node(), GraphBuilder::add_deconvolution_node(), GraphBuilder::add_depthwise_convolution_node(), GraphBuilder::add_fully_connected_layer(), GraphBuilder::add_normalize_planar_yuv_node(), GraphBuilder::add_scale_layer(), DepthToSpaceLayerNode::compute_output_descriptor(), SliceLayerNode::compute_output_descriptor(), ReorgLayerNode::compute_output_descriptor(), DetectionOutputLayerNode::compute_output_descriptor(), PoolingLayerNode::compute_output_descriptor(), PriorBoxLayerNode::compute_output_descriptor(), DeconvolutionLayerNode::compute_output_descriptor(), SplitLayerNode::compute_output_descriptor(), StridedSliceLayerNode::compute_output_descriptor(), FullyConnectedLayerNode::compute_output_descriptor(), FusedConvolutionBatchNormalizationNode::compute_output_descriptor(), DepthwiseConvolutionLayerNode::compute_output_descriptor(), FusedDepthwiseConvolutionBatchNormalizationNode::compute_output_descriptor(), ConvolutionLayerNode::compute_output_descriptor(), FullyConnectedLayerNode::compute_weights_descriptor(), FlattenLayerNode::configure_output(), ReshapeLayerNode::configure_output(), GenerateProposalsLayerNode::configure_output(), ResizeLayerNode::configure_output(), DummyNode::configure_output(), PermuteLayerNode::configure_output(), ROIAlignLayerNode::configure_output(), PadLayerNode::configure_output(), ReductionLayerNode::configure_output(), ArgMinMaxLayerNode::configure_output(), SplitLayerNode::configure_output(), EltwiseLayerNode::configure_output(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), arm_compute::graph::get_dimension_size(), SplitLayerSubTensorMutator::mutate(), DotGraphPrinter::print(), TensorDescriptor::set_shape(), and SplitLayerNode::validate().
Target target { Target::UNSPECIFIED } |
Target.
Definition at line 113 of file TensorDescriptor.h.
Referenced by arm_compute::graph::configure_tensor(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), arm_compute::graph::backends::detail::get_backing_tensor(), DepthConcatSubTensorMutator::mutate(), and SplitLayerSubTensorMutator::mutate().