Descriptor for data with one index dimension, one channel dimension and two spatial dimensions.
More...
#include <NvInfer.h>
Descriptor for data with one index dimension, one channel dimension and two spatial dimensions.
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. |
int& nvinfer1::DimsNCHW::c |
( |
| ) |
|
|
inline |
Get the channel count.
- Returns
- The channel count.
int nvinfer1::DimsNCHW::c |
( |
| ) |
const |
|
inline |
Get the channel count.
- Returns
- The channel count.
int& nvinfer1::DimsNCHW::h |
( |
| ) |
|
|
inline |
Get the height.
- Returns
- The height.
int nvinfer1::DimsNCHW::h |
( |
| ) |
const |
|
inline |
Get the height.
- Returns
- The height.
int& nvinfer1::DimsNCHW::n |
( |
| ) |
|
|
inline |
Get the index count.
- Returns
- The index count.
int nvinfer1::DimsNCHW::n |
( |
| ) |
const |
|
inline |
Get the index count.
- Returns
- The index count.
int& nvinfer1::DimsNCHW::w |
( |
| ) |
|
|
inline |
Get the width.
- Returns
- The width.
int nvinfer1::DimsNCHW::w |
( |
| ) |
const |
|
inline |
Get the width.
- Returns
- The width.
The documentation for this class was generated from the following file: