apollo::perception::inference::ReLUPlugin Class Reference

#include <leakyReLU_plugin.h>

Inheritance diagram for apollo::perception::inference::ReLUPlugin:

Collaboration diagram for apollo::perception::inference::ReLUPlugin:

Public Member Functions
	ReLUPlugin (const ReLUParameter &param, const nvinfer1::Dims &in_dims)
	ReLUPlugin ()
	~ReLUPlugin ()
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

ReLUPlugin() [1/2]

apollo::perception::inference::ReLUPlugin::ReLUPlugin ( const ReLUParameter &  param, const nvinfer1::Dims &  in_dims  )

inline

ReLUPlugin() [2/2]

apollo::perception::inference::ReLUPlugin::ReLUPlugin ( )

inline

~ReLUPlugin()

apollo::perception::inference::ReLUPlugin::~ReLUPlugin ( )

inline

Member Function Documentation

configure()

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

inlineoverride

enqueue()

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

virtual

getNbOutputs()

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

inlineoverride

getOutputDimensions()

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

inlineoverride

getSerializationSize()

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

inlineoverride

getWorkspaceSize()

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

inlineoverride

initialize()

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

inlinevirtual

serialize()

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

inlineoverride

terminate()

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

inlinevirtual

---

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

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