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