TensorInfo structure. More...

#include <TensorInfo.hpp>

Public Types
enum	DataType : int { DataType::FP16 = 0, DataType::U8F = 1, DataType::INT = 2, DataType::FP32 = 3, DataType::I8 = 4 }

enum	StorageOrder : int { StorageOrder::NHWC = 0x4213, StorageOrder::NHCW = 0x4231, StorageOrder::NCHW = 0x4321, StorageOrder::HWC = 0x213, StorageOrder::CHW = 0x321, StorageOrder::WHC = 0x123, StorageOrder::HCW = 0x231, StorageOrder::WCH = 0x132, StorageOrder::CWH = 0x312, StorageOrder::NC = 0x43, StorageOrder::CN = 0x34, StorageOrder::C = 0x3, StorageOrder::H = 0x2, StorageOrder::W = 0x1 }

Public Attributes
DataType	dataType = DataType::FP16

std::vector< unsigned >	dims

std::string	name

unsigned int	numDimensions = 0

unsigned int	offset = 0

StorageOrder	order = StorageOrder::NCHW

std::vector< unsigned >	strides

Detailed Description

TensorInfo structure.

Definition at line 12 of file TensorInfo.hpp.

Member Enumeration Documentation

◆ DataType

enum dai::TensorInfo::DataType : int

strong

Enumerator
FP16
U8F
INT
FP32
I8

Definition at line 30 of file TensorInfo.hpp.

◆ StorageOrder

enum dai::TensorInfo::StorageOrder : int

strong

Enumerator
NHWC
NHCW
NCHW
HWC
CHW
WHC
HCW
WCH
CWH
NC
CN
C
H
W

Definition at line 13 of file TensorInfo.hpp.

Member Data Documentation

◆ dataType

DataType dai::TensorInfo::dataType = DataType::FP16

Definition at line 39 of file TensorInfo.hpp.

◆ dims

std::vector<unsigned> dai::TensorInfo::dims

Definition at line 41 of file TensorInfo.hpp.

◆ name

std::string dai::TensorInfo::name

Definition at line 43 of file TensorInfo.hpp.

◆ numDimensions

unsigned int dai::TensorInfo::numDimensions = 0

Definition at line 40 of file TensorInfo.hpp.

◆ offset

unsigned int dai::TensorInfo::offset = 0

Definition at line 44 of file TensorInfo.hpp.

◆ order

StorageOrder dai::TensorInfo::order = StorageOrder::NCHW

Definition at line 38 of file TensorInfo.hpp.

◆ strides

std::vector<unsigned> dai::TensorInfo::strides

Definition at line 42 of file TensorInfo.hpp.

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

TensorInfo.hpp

Public Types

Public Attributes