attachToContext(cudnnContext *cudnnContext, cublasContext *cublasContext, IGpuAllocator *gpuAllocator) override | nvinfer1::plugin::GenerateDetection | virtual |
canBroadcastInputAcrossBatch(int inputIndex) const override | nvinfer1::plugin::GenerateDetection | |
nvinfer1::IPluginV2Ext::canBroadcastInputAcrossBatch(int32_t inputIndex) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
check_valid_inputs(const nvinfer1::Dims *inputs, int nbInputDims) | nvinfer1::plugin::GenerateDetection | private |
clone() const override | nvinfer1::plugin::GenerateDetection | virtual |
configurePlugin(const Dims *inputDims, int nbInputs, const Dims *outputDims, int nbOutputs, const DataType *inputTypes, const DataType *outputTypes, const bool *inputIsBroadcast, const bool *outputIsBroadcast, PluginFormat floatFormat, int maxBatchSize) override | nvinfer1::plugin::GenerateDetection | |
nvinfer1::IPluginV2Ext::configurePlugin(const Dims *inputDims, int32_t nbInputs, const Dims *outputDims, int32_t nbOutputs, const DataType *inputTypes, const DataType *outputTypes, const bool *inputIsBroadcast, const bool *outputIsBroadcast, PluginFormat floatFormat, int32_t maxBatchSize)=0 | nvinfer1::IPluginV2Ext | pure virtual |
configureWithFormat(const Dims *, int32_t, const Dims *, int32_t, DataType, PluginFormat, int32_t) | nvinfer1::IPluginV2Ext | inlineprotectedvirtual |
destroy() override | nvinfer1::plugin::GenerateDetection | virtual |
detachFromContext() override | nvinfer1::plugin::GenerateDetection | virtual |
enqueue(int batch_size, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream) override | nvinfer1::plugin::GenerateDetection | |
nvinfer1::IPluginV2Ext::enqueue(int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream)=0 | nvinfer1::IPluginV2 | pure virtual |
GenerateDetection(int num_classes, int keep_topk, float score_threshold, float iou_threshold, const nvinfer1::Dims &image_size) | nvinfer1::plugin::GenerateDetection | |
GenerateDetection(const void *data, size_t length) | nvinfer1::plugin::GenerateDetection | |
getNbOutputs() const override | nvinfer1::plugin::GenerateDetection | virtual |
getOutputDataType(int index, const nvinfer1::DataType *inputTypes, int nbInputs) const override | nvinfer1::plugin::GenerateDetection | |
nvinfer1::IPluginV2Ext::getOutputDataType(int32_t index, const nvinfer1::DataType *inputTypes, int32_t nbInputs) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
getOutputDimensions(int index, const Dims *inputs, int nbInputDims) override | nvinfer1::plugin::GenerateDetection | |
nvinfer1::IPluginV2Ext::getOutputDimensions(int32_t index, const Dims *inputs, int32_t nbInputDims)=0 | nvinfer1::IPluginV2 | pure virtual |
getPluginNamespace() const override | nvinfer1::plugin::GenerateDetection | virtual |
getPluginType() const override | nvinfer1::plugin::GenerateDetection | virtual |
getPluginVersion() const override | nvinfer1::plugin::GenerateDetection | virtual |
getSerializationSize() const override | nvinfer1::plugin::GenerateDetection | virtual |
getTensorRTVersion() const | nvinfer1::IPluginV2Ext | inlineprotectedvirtual |
getWorkspaceSize(int maxBatchSize) const override | nvinfer1::plugin::GenerateDetection | |
nvinfer1::IPluginV2Ext::getWorkspaceSize(int32_t maxBatchSize) const =0 | nvinfer1::IPluginV2 | pure virtual |
initialize() override | nvinfer1::plugin::GenerateDetection | virtual |
isOutputBroadcastAcrossBatch(int outputIndex, const bool *inputIsBroadcasted, int nbInputs) const override | nvinfer1::plugin::GenerateDetection | |
nvinfer1::IPluginV2Ext::isOutputBroadcastAcrossBatch(int32_t outputIndex, const bool *inputIsBroadcasted, int32_t nbInputs) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
mAnchorsCnt | nvinfer1::plugin::GenerateDetection | private |
mBackgroundLabel | nvinfer1::plugin::GenerateDetection | private |
mImageSize | nvinfer1::plugin::GenerateDetection | private |
mIOUThreshold | nvinfer1::plugin::GenerateDetection | private |
mKeepTopK | nvinfer1::plugin::GenerateDetection | private |
mMaxBatchSize | nvinfer1::plugin::GenerateDetection | private |
mNameSpace | nvinfer1::plugin::GenerateDetection | private |
mNbClasses | nvinfer1::plugin::GenerateDetection | private |
mParam | nvinfer1::plugin::GenerateDetection | private |
mRegWeightDevice | nvinfer1::plugin::GenerateDetection | private |
mScoreThreshold | nvinfer1::plugin::GenerateDetection | private |
mType | nvinfer1::plugin::GenerateDetection | private |
mValidCnt | nvinfer1::plugin::GenerateDetection | private |
serialize(void *buffer) const override | nvinfer1::plugin::GenerateDetection | virtual |
setPluginNamespace(const char *libNamespace) override | nvinfer1::plugin::GenerateDetection | virtual |
supportsFormat(DataType type, PluginFormat format) const override | nvinfer1::plugin::GenerateDetection | virtual |
terminate() override | nvinfer1::plugin::GenerateDetection | virtual |
~GenerateDetection() override=default | nvinfer1::plugin::GenerateDetection | |
~IPluginV2() | nvinfer1::IPluginV2 | inlineprotectedvirtual |
~IPluginV2Ext() | nvinfer1::IPluginV2Ext | inlinevirtual |