TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
BufferDesc Class Reference
Collaboration diagram for BufferDesc:

Public Member Functions

 BufferDesc ()=default
 
 BufferDesc (nvinfer1::Dims dims, int dataWidth, TensorFormat format)
 
int getElememtSize ()
 
int getBufferSize ()
 

Public Attributes

int dims [4] = {1, 1, 1, 1}
 
int dataWidth = 1
 
int scalarPerVector = 1
 
bool channelPivot = false
 

Constructor & Destructor Documentation

◆ BufferDesc() [1/2]

BufferDesc::BufferDesc ( )
default

◆ BufferDesc() [2/2]

BufferDesc::BufferDesc ( nvinfer1::Dims  dims,
int  dataWidth,
TensorFormat  format 
)
inline

Member Function Documentation

◆ getElememtSize()

int BufferDesc::getElememtSize ( )
inline
Here is the caller graph for this function:

◆ getBufferSize()

int BufferDesc::getBufferSize ( )
inline
Here is the call graph for this function:
Here is the caller graph for this function:

Member Data Documentation

◆ dims

int BufferDesc::dims[4] = {1, 1, 1, 1}

◆ dataWidth

int BufferDesc::dataWidth = 1

◆ scalarPerVector

int BufferDesc::scalarPerVector = 1

◆ channelPivot

bool BufferDesc::channelPivot = false

The documentation for this class was generated from the following file: