Descriptor for data with one index dimension, one channel dimension and two spatial dimensions.
More...
#include <NvInfer.h>
|
int | nbDims |
| The number of dimensions.
|
|
int | d [MAX_DIMS] |
| The extent of each dimension.
|
|
TRT_DEPRECATED DimensionType | type [MAX_DIMS] |
| The type of each dimension.
|
|
static const int | MAX_DIMS = 8 |
| The maximum number of dimensions supported for a tensor.
|
|
Descriptor for data with one index dimension, one channel dimension and two spatial dimensions.
- Deprecated:
- DimsNCHW will be removed in a future version of TensorRT, use Dims instead.
◆ DimsNCHW()
nvinfer1::DimsNCHW::DimsNCHW |
( |
int |
batchSize, |
|
|
int |
channels, |
|
|
int |
height, |
|
|
int |
width |
|
) |
| |
|
inline |
Construct a DimsNCHW given batch size, channel count, height and width.
- Parameters
-
batchSize | The batch size (commonly denoted N). |
channels | The channel count. |
height | The height of the data. |
width | The width of the data. |
◆ c() [1/2]
int& nvinfer1::DimsNCHW::c |
( |
| ) |
|
|
inline |
Get the channel count.
- Returns
- The channel count.
◆ c() [2/2]
int nvinfer1::DimsNCHW::c |
( |
| ) |
const |
|
inline |
Get the channel count.
- Returns
- The channel count.
◆ h() [1/2]
int& nvinfer1::DimsNCHW::h |
( |
| ) |
|
|
inline |
Get the height.
- Returns
- The height.
◆ h() [2/2]
int nvinfer1::DimsNCHW::h |
( |
| ) |
const |
|
inline |
Get the height.
- Returns
- The height.
◆ n() [1/2]
int& nvinfer1::DimsNCHW::n |
( |
| ) |
|
|
inline |
Get the index count.
- Returns
- The index count.
◆ n() [2/2]
int nvinfer1::DimsNCHW::n |
( |
| ) |
const |
|
inline |
Get the index count.
- Returns
- The index count.
◆ w() [1/2]
int& nvinfer1::DimsNCHW::w |
( |
| ) |
|
|
inline |
Get the width.
- Returns
- The width.
◆ w() [2/2]
int nvinfer1::DimsNCHW::w |
( |
| ) |
const |
|
inline |
Get the width.
- Returns
- The width.
The documentation for this class was generated from the following file: