TensorRT  7.0.0.11
nvinfer1::DimsNCHW Class Reference

Descriptor for data with one index dimension, one channel dimension and two spatial dimensions. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::DimsNCHW:
nvinfer1::Dims4 nvinfer1::Dims

Public Member Functions

 DimsNCHW ()
 Construct an empty DimsNCHW object.
 
 DimsNCHW (int batchSize, int channels, int height, int width)
 Construct a DimsNCHW given batch size, channel count, height and width. More...
 
int & n ()
 Get the index count. More...
 
int n () const
 Get the index count. More...
 
int & c ()
 Get the channel count. More...
 
int c () const
 Get the channel count. More...
 
int & h ()
 Get the height. More...
 
int h () const
 Get the height. More...
 
int & w ()
 Get the width. More...
 
int w () const
 Get the width. More...
 
- Public Member Functions inherited from nvinfer1::Dims4
 Dims4 ()
 Construct an empty Dims2 object.
 
 Dims4 (int d0, int d1, int d2, int d3)
 Construct a Dims4 from 4 elements. More...
 

Additional Inherited Members

- Public Attributes inherited from nvinfer1::Dims
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 Public Attributes inherited from nvinfer1::Dims
static const int MAX_DIMS = 8
 The maximum number of dimensions supported for a tensor.
 

Detailed Description

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.

Constructor & Destructor Documentation

nvinfer1::DimsNCHW::DimsNCHW ( int  batchSize,
int  channels,
int  height,
int  width 
)
inline

Construct a DimsNCHW given batch size, channel count, height and width.

Parameters
batchSizeThe batch size (commonly denoted N).
channelsThe channel count.
heightThe height of the data.
widthThe width of the data.

Member Function Documentation

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: