24.02.1
|
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 |
Public Member Functions inherited from ICloneable< TensorDescriptor > | |
virtual | ~ICloneable ()=default |
Default virtual desctructor. More... | |
virtual std::unique_ptr< TensorDescriptor > | clone () const=0 |
Provide a clone of the current object of class T. 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 |
|
inlineoverride |
Definition at line 108 of file TensorDescriptor.h.
|
inline |
Sets tensor descriptor data type.
[in] | tensor_data_type | Data type |
Definition at line 79 of file TensorDescriptor.h.
References TensorDescriptor::data_type.
|
inline |
Sets tensor descriptor data layout.
[in] | data_layout | Data layout |
Definition at line 90 of file TensorDescriptor.h.
References arm_compute::cpu::data_layout, and TensorDescriptor::layout.
|
inline |
Sets tensor descriptor quantization info.
[in] | tensor_quant_info | Quantization information |
Definition at line 101 of file TensorDescriptor.h.
References TensorDescriptor::quant_info.
|
inline |
Sets tensor descriptor shape.
[in] | tensor_shape | Tensor shape to set |
Definition at line 68 of file TensorDescriptor.h.
References TensorDescriptor::shape.
DataType data_type {DataType::UNKNOWN} |
Data type.
Definition at line 114 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(), DetectionPostProcessLayerNode::configure_output(), SoftmaxLayerNode::configure_output(), GenerateProposalsLayerNode::configure_output(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), and TensorDescriptor::set_data_type().
DataLayout layout {DataLayout::NCHW} |
Data layout.
Definition at line 115 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(), ConcatenateLayerNode::compute_output_descriptor(), DepthToSpaceLayerNode::compute_output_descriptor(), ReorgLayerNode::compute_output_descriptor(), PoolingLayerNode::compute_output_descriptor(), DeconvolutionLayerNode::compute_output_descriptor(), FusedConvolutionBatchNormalizationNode::compute_output_descriptor(), FusedDepthwiseConvolutionBatchNormalizationNode::compute_output_descriptor(), DepthwiseConvolutionLayerNode::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::detail::fuse_pad_with_convolution(), arm_compute::graph::get_dimension_size(), and TensorDescriptor::set_layout().
QuantizationInfo quant_info {} |
Quantization info.
Definition at line 116 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(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), NodeFusionMutator::mutate(), and TensorDescriptor::set_quantization_info().
TensorShape shape {} |
Tensor shape.
Definition at line 113 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(), StackLayerNode::compute_output_descriptor(), ConcatenateLayerNode::compute_output_descriptor(), 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(), FusedConvolutionBatchNormalizationNode::compute_output_descriptor(), FullyConnectedLayerNode::compute_output_descriptor(), FusedDepthwiseConvolutionBatchNormalizationNode::compute_output_descriptor(), DepthwiseConvolutionLayerNode::compute_output_descriptor(), ConvolutionLayerNode::compute_output_descriptor(), FullyConnectedLayerNode::compute_weights_descriptor(), FlattenLayerNode::configure_output(), ReshapeLayerNode::configure_output(), DetectionPostProcessLayerNode::configure_output(), GenerateProposalsLayerNode::configure_output(), ResizeLayerNode::configure_output(), PermuteLayerNode::configure_output(), DummyNode::configure_output(), ROIAlignLayerNode::configure_output(), PadLayerNode::configure_output(), SplitLayerNode::configure_output(), EltwiseLayerNode::configure_output(), NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), arm_compute::graph::get_dimension_size(), SplitLayerSubTensorMutator::mutate(), and TensorDescriptor::set_shape().
Target target {Target::UNSPECIFIED} |
Target.
Definition at line 117 of file TensorDescriptor.h.
Referenced by NEDeviceBackend::create_tensor(), CLDeviceBackend::create_tensor(), and SplitLayerSubTensorMutator::mutate().