TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
Batch Struct Reference
Collaboration diagram for Batch:

Public Member Functions

 Batch (ICudaEngine *engine, void *userInputPtr, void *itemInputPtr, const Args &args)
 
 ~Batch ()
 

Public Attributes

ICudaEnginemEngine
 
SampleUniquePtr< nvinfer1::IExecutionContextmContext
 
cudaStream_t mStream
 
void * mHostMemory [5]
 
void * mDeviceMemory [5]
 
std::vector< size_t > mMemSizes
 

Constructor & Destructor Documentation

◆ Batch()

Batch::Batch ( ICudaEngine engine,
void *  userInputPtr,
void *  itemInputPtr,
const Args args 
)
inline

◆ ~Batch()

Batch::~Batch ( )
inline

Member Data Documentation

◆ mEngine

ICudaEngine* Batch::mEngine

◆ mContext

SampleUniquePtr<nvinfer1::IExecutionContext> Batch::mContext

◆ mStream

cudaStream_t Batch::mStream

◆ mHostMemory

void* Batch::mHostMemory[5]

◆ mDeviceMemory

void* Batch::mDeviceMemory[5]

◆ mMemSizes

std::vector<size_t> Batch::mMemSizes

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