24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_TENSOR_DESCRIPTOR_H
25 #define ARM_COMPUTE_GRAPH_TENSOR_DESCRIPTOR_H
55 :
shape(tensor_shape),
57 layout(tensor_data_layout),
108 std::unique_ptr<TensorDescriptor>
clone()
const override
110 return std::make_unique<TensorDescriptor>(*
this);
@ NCHW
Num samples, channels, height, width.
TensorDescriptor & set_data_type(DataType tensor_data_type)
Sets tensor descriptor data type.
TensorDescriptor()=default
Default Constructor.
Quantization information.
DataLayout
[DataLayout enum definition]
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.
QuantizationInfo quant_info
Quantization info.
constexpr auto data_layout
DataLayout layout
Data layout.
@ UNSPECIFIED
Unspecified Target.
TensorShape shape
Tensor shape.
Copyright (c) 2017-2024 Arm Limited.
TensorDescriptor & set_layout(DataLayout data_layout)
Sets tensor descriptor data layout.
TensorDescriptor & set_shape(TensorShape &tensor_shape)
Sets tensor descriptor shape.
DataType data_type
Data type.
@ UNKNOWN
Unknown data type.
TensorDescriptor & set_quantization_info(QuantizationInfo tensor_quant_info)
Sets tensor descriptor quantization info.
std::unique_ptr< TensorDescriptor > clone() const override
DataType
Available data types.