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() const
inline int getDataTypeSize() const
inline int getWidth() const
inline int getHeight() const
inline std::size_t getTensorSize() const
inline int getChannels() const
inline size_t getChannelStride() const
inline size_t getWidthStride() const
inline size_t getHeightStride() const

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