|
VART-X
0.3.0
|
Tensor config information. More...
#include <vart_postprocess_types.hpp>
Public Attributes | |
| uint32_t | size |
| float | scale_coeff |
| std::vector< uint32_t > | shape |
| std::string | name |
| TensorDataType | data_type |
| TensorDataDirection | direction |
Tensor config information.
| vart::TensorInfo::data_type |
Element data type of the tensor (see TensorDataType).
| vart::TensorInfo::direction |
Whether this tensor is an input or output of the model (see TensorDataDirection).
| vart::TensorInfo::name |
Name of the tensor
| vart::TensorInfo::scale_coeff |
Quantization scale factor of tensor
| vart::TensorInfo::shape |
Shape dimension of each tensor
| vart::TensorInfo::size |
Size of the entire tensor in bytes