nvinfer2::safe::PhysicalDims Class Reference

Structure to define the physical dimensions of a tensor with support for up to 9 dimensions. More...

#include <NvInferSafeRuntime.h>

Public Attributes
int32_t	nbDims
	The rank (number of dimensions). More...
int64_t	d [MAX_DIMS]
	The extent of each dimension. More...

Static Public Attributes
static constexpr int32_t	MAX_DIMS {9}

Detailed Description

Structure to define the physical dimensions of a tensor with support for up to 9 dimensions.

This structure is specifically designed for describing physical tensors layout in memory which can have up to 9 dimensions after vectorization. Unlike the standard Dims structure which is limited to 8 dimensions, PhysicalDims supports the additional dimension needed when tensors are vectorized.

When a tensor is vectorized by size T along some dimension with size K, the vectorized dimension is split into two dimensions of sizes ceil(K/T) and T, potentially resulting in 9 physical dimensions for an originally 8 logical dimensional tensor.

Member Data Documentation

d

int64_t nvinfer2::safe::PhysicalDims::d[MAX_DIMS]

The extent of each dimension.

MAX_DIMS

constexpr int32_t nvinfer2::safe::PhysicalDims::MAX_DIMS {9}

staticconstexpr

The maximum rank (number of dimensions) supported for physical tensor layout. This is set to 9 to accommodate vectorized tensors that split one dimension into two.

nbDims

int32_t nvinfer2::safe::PhysicalDims::nbDims

The rank (number of dimensions).

---

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

• NvInferSafeRuntime.h