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) | BufferManager | inline |
BufferManager(std::shared_ptr< nvinfer1::ICudaEngine > engine, int32_t const batchSize=0, nvinfer1::IExecutionContext const *context=nullptr) | BufferManager | inline |
copyInputToDevice() | BufferManager | inline |
copyInputToDeviceAsync(cudaStream_t const &stream=0) | BufferManager | inline |
copyOutputToHost() | BufferManager | inline |
copyOutputToHostAsync(cudaStream_t const &stream=0) | BufferManager | inline |
getBuffer(bool const isHost, std::string const &tensorName) const | BufferManager | inlineprivate |
getDeviceBindings() | BufferManager | inline |
getDeviceBindings() const | BufferManager | inline |
getDeviceBuffer(std::string const &tensorName) const | BufferManager | inline |
getHostBuffer(std::string const &tensorName) const | BufferManager | inline |
kINVALID_SIZE_VALUE | BufferManager | static |
mBatchSize | BufferManager | private |
mDeviceBindings | BufferManager | private |
memcpyBuffers(bool const copyInput, bool const deviceToHost, bool const async, cudaStream_t const &stream=0) | BufferManager | inlineprivate |
mEngine | BufferManager | private |
mManagedBuffers | BufferManager | private |
mNames | BufferManager | private |
print(std::ostream &os, void *buf, size_t bufSize, size_t rowCount) | BufferManager | inline |
size(std::string const &tensorName) const | BufferManager | inline |
tenosrIsInput(const std::string &tensorName) const | BufferManager | inlineprivate |
~BufferManager()=default | BufferManager |