apollo::perception::inference::DFMBPSROIAlignPlugin Class Reference

#include <dfmb_psroi_align_plugin.h>

Inheritance diagram for apollo::perception::inference::DFMBPSROIAlignPlugin:

Collaboration diagram for apollo::perception::inference::DFMBPSROIAlignPlugin:

Public Member Functions
	DFMBPSROIAlignPlugin (const DFMBPSROIAlignParameter &dfmb_psroi_align_parameter, nvinfer1::Dims *in_dims, int nbInputs)
virtual	~DFMBPSROIAlignPlugin ()
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

DFMBPSROIAlignPlugin()

apollo::perception::inference::DFMBPSROIAlignPlugin::DFMBPSROIAlignPlugin ( const DFMBPSROIAlignParameter &  dfmb_psroi_align_parameter, nvinfer1::Dims *  in_dims, int  nbInputs  )

inline

~DFMBPSROIAlignPlugin()

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

inlinevirtual

Member Function Documentation

configure()

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

inlineoverride

enqueue()

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

virtual

getNbOutputs()

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

inlineoverride

getOutputDimensions()

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

inlineoverride

getSerializationSize()

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

inlineoverride

getWorkspaceSize()

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

inlineoverride

initialize()

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

inlinevirtual

serialize()

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

inlineoverride

terminate()

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

inlinevirtual

---

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

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