TensorRT
|
structure to define the dimensions of a tensor More...
#include <NvInfer.h>
Public Attributes | |
int | nbDims |
the number of dimensions | |
int | d [MAX_DIMS] |
the extent of each dimension | |
DimensionType | type [MAX_DIMS] |
the type of each dimension | |
Static Public Attributes | |
static const int | MAX_DIMS = 8 |
the maximum number of dimensions supported for a tensor | |
structure to define the dimensions of a tensor