configure(const nvinfer1::Dims *inputDims, int nbInputs, const nvinfer1::Dims *outputDims, int nbOutputs, int maxBatchSize) override | nmtSample::DebugUtil::DumpTensorPlugin | |
nvinfer1::IPlugin::configure(const Dims *inputDims, int32_t nbInputs, const Dims *outputDims, int32_t nbOutputs, int32_t maxBatchSize)=0 | nvinfer1::IPlugin | pure virtual |
DumpTensorPlugin(std::shared_ptr< std::ostream > out) | nmtSample::DebugUtil::DumpTensorPlugin | |
enqueue(int batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream) override | nmtSample::DebugUtil::DumpTensorPlugin | |
nvinfer1::IPlugin::enqueue(int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream)=0 | nvinfer1::IPlugin | pure virtual |
getNbOutputs() const override | nmtSample::DebugUtil::DumpTensorPlugin | virtual |
getOutputDimensions(int index, const nvinfer1::Dims *inputs, int nbInputDims) override | nmtSample::DebugUtil::DumpTensorPlugin | |
nvinfer1::IPlugin::getOutputDimensions(int32_t index, const Dims *inputs, int32_t nbInputDims)=0 | nvinfer1::IPlugin | pure virtual |
getSerializationSize() override | nmtSample::DebugUtil::DumpTensorPlugin | virtual |
getWorkspaceSize(int maxBatchSize) const override | nmtSample::DebugUtil::DumpTensorPlugin | |
nvinfer1::IPlugin::getWorkspaceSize(int32_t maxBatchSize) const =0 | nvinfer1::IPlugin | pure virtual |
initialize() override | nmtSample::DebugUtil::DumpTensorPlugin | virtual |
mData | nmtSample::DebugUtil::DumpTensorPlugin | private |
mDims | nmtSample::DebugUtil::DumpTensorPlugin | private |
mElemsPerRow | nmtSample::DebugUtil::DumpTensorPlugin | private |
mOut | nmtSample::DebugUtil::DumpTensorPlugin | private |
mTensorVolume | nmtSample::DebugUtil::DumpTensorPlugin | private |
ptr typedef | nmtSample::DebugUtil::DumpTensorPlugin | |
serialize(void *buffer) override | nmtSample::DebugUtil::DumpTensorPlugin | virtual |
terminate() override | nmtSample::DebugUtil::DumpTensorPlugin | virtual |
~DumpTensorPlugin() override=default | nmtSample::DebugUtil::DumpTensorPlugin | |
~IPlugin() | nvinfer1::IPlugin | inlinevirtual |