Struct TensorInfo

Struct Documentation

struct TensorInfo

TensorInfo structure.

Public Types

enum class StorageOrder : int

Values:

enumerator NHWC
enumerator NHCW
enumerator NCHW
enumerator HWC
enumerator CHW
enumerator WHC
enumerator HCW
enumerator WCH
enumerator CWH
enumerator NC
enumerator CN
enumerator C
enumerator H
enumerator W
enum class DataType : int

Values:

enumerator FP16
enumerator U8F
enumerator INT
enumerator FP32
enumerator I8
enumerator FP64

Public Functions

inline void validateStorageOrder()
inline int getDataTypeSize()
inline int getWidth()
inline int getHeight()
inline std::size_t getTensorSize()
inline int getChannels()

Public Members

StorageOrder order = StorageOrder::NCHW
DataType dataType = DataType::FP16
unsigned int numDimensions = 0
std::vector<unsigned> dims
std::vector<unsigned> strides
std::string name
unsigned int offset = 0
bool quantization = false
float qpScale = 1
float qpZp = 0