apollo::perception::inference::RCNNProposalPlugin Class Reference

#include <rcnn_proposal_plugin.h>

Inheritance diagram for apollo::perception::inference::RCNNProposalPlugin:

Collaboration diagram for apollo::perception::inference::RCNNProposalPlugin:

Public Member Functions
	RCNNProposalPlugin (const BBoxRegParameter &bbox_reg_param, const DetectionOutputSSDParameter &detection_output_ssd_param, nvinfer1::Dims *in_dims)
virtual	~RCNNProposalPlugin ()
virtual int	initialize ()
virtual void	terminate ()
int	getNbOutputs () const override
nvinfer1::Dims	getOutputDimensions (int index, const nvinfer1::Dims *inputs, int nbInputDims) override
void	configure (const nvinfer1::Dims *inputDims, int nbInputs, const nvinfer1::Dims *outputDims, int nbOutputs, int maxBatchSize) override
size_t	getWorkspaceSize (int maxBatchSize) const override
virtual int	enqueue (int batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream)
size_t	getSerializationSize () override
void	serialize (void *buffer) override

Constructor & Destructor Documentation

RCNNProposalPlugin()

apollo::perception::inference::RCNNProposalPlugin::RCNNProposalPlugin ( const BBoxRegParameter &  bbox_reg_param, const DetectionOutputSSDParameter &  detection_output_ssd_param, nvinfer1::Dims *  in_dims  )

inline

~RCNNProposalPlugin()

virtual apollo::perception::inference::RCNNProposalPlugin::~RCNNProposalPlugin ( )

inlinevirtual

Member Function Documentation

configure()

void apollo::perception::inference::RCNNProposalPlugin::configure ( const nvinfer1::Dims *  inputDims, int  nbInputs, const nvinfer1::Dims *  outputDims, int  nbOutputs, int  maxBatchSize  )

inlineoverride

enqueue()

virtual int apollo::perception::inference::RCNNProposalPlugin::enqueue ( int  batchSize, const void *const *  inputs, void **  outputs, void *  workspace, cudaStream_t  stream  )

virtual

getNbOutputs()

int apollo::perception::inference::RCNNProposalPlugin::getNbOutputs ( ) const

inlineoverride

getOutputDimensions()

nvinfer1::Dims apollo::perception::inference::RCNNProposalPlugin::getOutputDimensions ( int  index, const nvinfer1::Dims *  inputs, int  nbInputDims  )

inlineoverride

getSerializationSize()

size_t apollo::perception::inference::RCNNProposalPlugin::getSerializationSize ( )

inlineoverride

getWorkspaceSize()

size_t apollo::perception::inference::RCNNProposalPlugin::getWorkspaceSize ( int  maxBatchSize ) const

inlineoverride

initialize()

virtual int apollo::perception::inference::RCNNProposalPlugin::initialize ( )

inlinevirtual

serialize()

void apollo::perception::inference::RCNNProposalPlugin::serialize ( void *  buffer )

inlineoverride

terminate()

virtual void apollo::perception::inference::RCNNProposalPlugin::terminate ( )

inlinevirtual

---

The documentation for this class was generated from the following file:

• modules/perception/inference/tensorrt/plugins/rcnn_proposal_plugin.h