API
 
Loading...
Searching...
No Matches
BufferManager Member List

This is the complete list of members for BufferManager, including all inherited members.

BufferManager(std::shared_ptr< nvinfer1::ICudaEngine > engine, std::vector< int64_t > const &volumes, int32_t batchSize=0)BufferManagerinline
BufferManager(std::shared_ptr< nvinfer1::ICudaEngine > engine, int32_t const batchSize=0, nvinfer1::IExecutionContext const *context=nullptr)BufferManagerinline
copyInputToDevice()BufferManagerinline
copyInputToDeviceAsync(cudaStream_t const &stream=0)BufferManagerinline
copyOutputToHost()BufferManagerinline
copyOutputToHostAsync(cudaStream_t const &stream=0)BufferManagerinline
getBuffer(bool const isHost, std::string const &tensorName) constBufferManagerinlineprivate
getDeviceBindings()BufferManagerinline
getDeviceBindings() constBufferManagerinline
getDeviceBuffer(std::string const &tensorName) constBufferManagerinline
getHostBuffer(std::string const &tensorName) constBufferManagerinline
kINVALID_SIZE_VALUEBufferManagerstatic
mBatchSizeBufferManagerprivate
mDeviceBindingsBufferManagerprivate
memcpyBuffers(bool const copyInput, bool const deviceToHost, bool const async, cudaStream_t const &stream=0)BufferManagerinlineprivate
mEngineBufferManagerprivate
mManagedBuffersBufferManagerprivate
mNamesBufferManagerprivate
print(std::ostream &os, void *buf, size_t bufSize, size_t rowCount)BufferManagerinline
size(std::string const &tensorName) constBufferManagerinline
tenosrIsInput(const std::string &tensorName) constBufferManagerinlineprivate
~BufferManager()=defaultBufferManager