apollo::perception::inference::SoftmaxPlugin Class Reference

#include <softmax_plugin.h>

Inheritance diagram for apollo::perception::inference::SoftmaxPlugin:

Collaboration diagram for apollo::perception::inference::SoftmaxPlugin:

Public Member Functions
	SoftmaxPlugin (const SoftmaxParameter &param, nvinfer1::Dims in_dims)
	SoftmaxPlugin ()
	~SoftmaxPlugin ()
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
int	enqueue (int batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream) override
size_t	getSerializationSize () override
void	serialize (void *buffer) override

Constructor & Destructor Documentation

SoftmaxPlugin() [1/2]

apollo::perception::inference::SoftmaxPlugin::SoftmaxPlugin ( const SoftmaxParameter &  param, nvinfer1::Dims  in_dims  )

inline

SoftmaxPlugin() [2/2]

apollo::perception::inference::SoftmaxPlugin::SoftmaxPlugin ( )

inline

~SoftmaxPlugin()

apollo::perception::inference::SoftmaxPlugin::~SoftmaxPlugin ( )

inline

Member Function Documentation

configure()

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

inlineoverride

enqueue()

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

override

getNbOutputs()

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

inlineoverride

getOutputDimensions()

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

inlineoverride

getSerializationSize()

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

inlineoverride

getWorkspaceSize()

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

inlineoverride

initialize()

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

inlinevirtual

serialize()

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

inlineoverride

terminate()

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

inlinevirtual

---

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

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