TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
sample::MirroredBuffer Class Reference

Coupled host and device buffers. More...

Collaboration diagram for sample::MirroredBuffer:

Public Member Functions

void allocate (size_t size)
 
void * getDeviceBuffer () const
 
void * getHostBuffer () const
 
void hostToDevice (TrtCudaStream &stream)
 
void deviceToHost (TrtCudaStream &stream)
 
size_t getSize () const
 

Private Attributes

size_t mSize {0}
 
TrtHostBuffer mHostBuffer
 
TrtDeviceBuffer mDeviceBuffer
 

Detailed Description

Coupled host and device buffers.

Member Function Documentation

◆ allocate()

void sample::MirroredBuffer::allocate ( size_t  size)
inline
Here is the call graph for this function:
Here is the caller graph for this function:

◆ getDeviceBuffer()

void* sample::MirroredBuffer::getDeviceBuffer ( ) const
inline
Here is the call graph for this function:

◆ getHostBuffer()

void* sample::MirroredBuffer::getHostBuffer ( ) const
inline
Here is the call graph for this function:
Here is the caller graph for this function:

◆ hostToDevice()

void sample::MirroredBuffer::hostToDevice ( TrtCudaStream stream)
inline
Here is the call graph for this function:

◆ deviceToHost()

void sample::MirroredBuffer::deviceToHost ( TrtCudaStream stream)
inline
Here is the call graph for this function:

◆ getSize()

size_t sample::MirroredBuffer::getSize ( ) const
inline
Here is the caller graph for this function:

Member Data Documentation

◆ mSize

size_t sample::MirroredBuffer::mSize {0}
private

◆ mHostBuffer

TrtHostBuffer sample::MirroredBuffer::mHostBuffer
private

◆ mDeviceBuffer

TrtDeviceBuffer sample::MirroredBuffer::mDeviceBuffer
private

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