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:

Public Member Functions
	DimsNCHW ()
	construct an empty DimsNCHW object
	DimsNCHW (int batchSize, int channels, int height, int width)
	construct a DimsCHW given 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...

Additional Inherited Members
Public Attributes inherited from nvinfer1::Dims
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 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

Constructor & Destructor Documentation

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

inline

construct a DimsCHW given 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

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:

• NvInfer.h