Descriptor for data with one index dimension, one channel dimension and two spatial dimensions.
More...
#include <NvInfer.h>
|
| DimsNCHW () |
| Construct an empty DimsNCHW object.
|
|
| DimsNCHW (int32_t batchSize, int32_t channels, int32_t height, int32_t width) |
| Construct a DimsNCHW given batch size, channel count, height and width. More...
|
|
int32_t & | n () |
| Get the index count. More...
|
|
int32_t | n () const |
| Get the index count. More...
|
|
int32_t & | c () |
| Get the channel count. More...
|
|
int32_t | c () const |
| Get the channel count. More...
|
|
int32_t & | h () |
| Get the height. More...
|
|
int32_t | h () const |
| Get the height. More...
|
|
int32_t & | w () |
| Get the width. More...
|
|
int32_t | w () const |
| Get the width. More...
|
|
| Dims4 () |
| Construct an empty Dims2 object.
|
|
| Dims4 (int32_t d0, int32_t d1, int32_t d2, int32_t d3) |
| Construct a Dims4 from 4 elements. More...
|
|
|
int32_t | nbDims |
| The number of dimensions.
|
|
int32_t | d [MAX_DIMS] |
| The extent of each dimension.
|
|
TRT_DEPRECATED DimensionType | type [MAX_DIMS] |
| The type of each dimension.
|
|
static const int32_t | 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 |
( |
int32_t |
batchSize, |
|
|
int32_t |
channels, |
|
|
int32_t |
height, |
|
|
int32_t |
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]
int32_t& nvinfer1::DimsNCHW::c |
( |
| ) |
|
|
inline |
Get the channel count.
- Returns
- The channel count.
◆ c() [2/2]
int32_t nvinfer1::DimsNCHW::c |
( |
| ) |
const |
|
inline |
Get the channel count.
- Returns
- The channel count.
◆ h() [1/2]
int32_t& nvinfer1::DimsNCHW::h |
( |
| ) |
|
|
inline |
Get the height.
- Returns
- The height.
◆ h() [2/2]
int32_t nvinfer1::DimsNCHW::h |
( |
| ) |
const |
|
inline |
Get the height.
- Returns
- The height.
◆ n() [1/2]
int32_t& nvinfer1::DimsNCHW::n |
( |
| ) |
|
|
inline |
Get the index count.
- Returns
- The index count.
◆ n() [2/2]
int32_t nvinfer1::DimsNCHW::n |
( |
| ) |
const |
|
inline |
Get the index count.
- Returns
- The index count.
◆ w() [1/2]
int32_t& nvinfer1::DimsNCHW::w |
( |
| ) |
|
|
inline |
Get the width.
- Returns
- The width.
◆ w() [2/2]
int32_t nvinfer1::DimsNCHW::w |
( |
| ) |
const |
|
inline |
Get the width.
- Returns
- The width.
The documentation for this class was generated from the following file: