apollo::perception::inference::ArgMax1Plugin Class Reference

#include <argmax_plugin.h>

Inheritance diagram for apollo::perception::inference::ArgMax1Plugin:

Collaboration diagram for apollo::perception::inference::ArgMax1Plugin:

Public Member Functions
	ArgMax1Plugin (const ArgMaxParameter &argmax_param, nvinfer1::Dims in_dims)
virtual int	initialize ()
	get the number of outputs from the layer More...
virtual void	terminate ()
int	getNbOutputs () const override
virtual nvinfer1::Dims	getOutputDimensions (int index, const nvinfer1::Dims *inputs, int nbInputDims)
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
virtual	~ArgMax1Plugin ()

Constructor & Destructor Documentation

ArgMax1Plugin()

apollo::perception::inference::ArgMax1Plugin::ArgMax1Plugin ( const ArgMaxParameter &  argmax_param, nvinfer1::Dims  in_dims  )

inline

~ArgMax1Plugin()

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

inlinevirtual

Member Function Documentation

configure()

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

inlineoverride

enqueue()

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

virtual

getNbOutputs()

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

inlineoverride

getOutputDimensions()

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

inlinevirtual

getSerializationSize()

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

inlineoverride

getWorkspaceSize()

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

inlineoverride

initialize()

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

inlinevirtual

get the number of outputs from the layer

Returns

the number of outputs

this function is called by the implementations of INetworkDefinition and IBuilder. In particular, it is called prior to any call to initialize().

serialize()

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

inlineoverride

terminate()

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

inlinevirtual

---

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

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