IPlugin类 参考

#include <plugin.h>

类 IPlugin 继承关系图:

IPlugin 的协作图:

Public 成员函数
virtual int32_t	getNbOutputs () const =0
virtual Dims	getOutputDimensions (int32_t index, const Dims *inputs, int32_t nbInputDims)=0
virtual void	configure (const Dims *inputDims, int32_t nbInputs, const Dims *outputDims, int32_t nbOutputs, int32_t maxBatchSize)=0
virtual int32_t	initialize ()=0
virtual void	terminate ()=0
virtual size_t	getWorkspaceSize (int32_t maxBatchSize) const =0
virtual int32_t	enqueue (int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream)=0
virtual size_t	getSerializationSize ()=0
virtual void	serialize (void *buffer)=0
virtual	~IPlugin ()

详细描述

在文件 plugin.h 第 46 行定义.

构造及析构函数说明

~IPlugin()

virtual IPlugin::~IPlugin ( )

inlinevirtual

在文件 plugin.h 第 62 行定义.

{}

成员函数说明

configure()

virtual void IPlugin::configure ( const Dims *  inputDims, int32_t  nbInputs, const Dims *  outputDims, int32_t  nbOutputs, int32_t  maxBatchSize  )

pure virtual

enqueue()

virtual int32_t IPlugin::enqueue ( int32_t  batchSize, const void *const *  inputs, void **  outputs, void *  workspace, cudaStream_t  stream  )

pure virtual

getNbOutputs()

virtual int32_t IPlugin::getNbOutputs ( ) const

pure virtual

在 apollo::perception::inference::DFMBPSROIAlignPlugin, apollo::perception::inference::RCNNProposalPlugin , 以及 apollo::perception::inference::RPNProposalSSDPlugin 内被实现.

getOutputDimensions()

virtual Dims IPlugin::getOutputDimensions ( int32_t  index, const Dims *  inputs, int32_t  nbInputDims  )

pure virtual

getSerializationSize()

virtual size_t IPlugin::getSerializationSize ( )

pure virtual

在 apollo::perception::inference::DFMBPSROIAlignPlugin, apollo::perception::inference::RCNNProposalPlugin , 以及 apollo::perception::inference::RPNProposalSSDPlugin 内被实现.

getWorkspaceSize()

virtual size_t IPlugin::getWorkspaceSize ( int32_t  maxBatchSize ) const

pure virtual

initialize()

virtual int32_t IPlugin::initialize ( )

pure virtual

在 apollo::perception::inference::DFMBPSROIAlignPlugin, apollo::perception::inference::RCNNProposalPlugin , 以及 apollo::perception::inference::RPNProposalSSDPlugin 内被实现.

serialize()

virtual void IPlugin::serialize ( void *  buffer )

pure virtual

在 apollo::perception::inference::DFMBPSROIAlignPlugin, apollo::perception::inference::RCNNProposalPlugin , 以及 apollo::perception::inference::RPNProposalSSDPlugin 内被实现.

terminate()

virtual void IPlugin::terminate ( )

pure virtual

在 apollo::perception::inference::DFMBPSROIAlignPlugin, apollo::perception::inference::RCNNProposalPlugin , 以及 apollo::perception::inference::RPNProposalSSDPlugin 内被实现.

---

该类的文档由以下文件生成:

• modules/perception/common/inference/migraphx/plugins/plugin.h