VART-X  0.3.0
vart::TensorInfo Struct Reference

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
 

Detailed Description

Tensor config information.

Member Data Documentation

◆ data_type

vart::TensorInfo::data_type

Element data type of the tensor (see TensorDataType).

◆ direction

vart::TensorInfo::direction

Whether this tensor is an input or output of the model (see TensorDataDirection).

◆ name

vart::TensorInfo::name

Name of the tensor

◆ scale_coeff

vart::TensorInfo::scale_coeff

Quantization scale factor of tensor

◆ shape

vart::TensorInfo::shape

Shape dimension of each tensor

◆ size

vart::TensorInfo::size

Size of the entire tensor in bytes


The documentation for this struct was generated from the following file: