Struct TensorInfo

Defined in File TensorInfo.hpp

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