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() 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