bert::QKVToContextInterleavedPluginCreator Class Reference

Inheritance diagram for bert::QKVToContextInterleavedPluginCreator:

Collaboration diagram for bert::QKVToContextInterleavedPluginCreator:

Public Member Functions
	QKVToContextInterleavedPluginCreator ()
const char *	getPluginName () const override
	Return the plugin name. More...
const char *	getPluginVersion () const override
	Return the plugin version. More...
const nvinfer1::PluginFieldCollection *	getFieldNames () override
	Return a list of fields that needs to be passed to createPlugin. More...
nvinfer1::IPluginV2 *	createPlugin (const char *name, const nvinfer1::PluginFieldCollection *fc) override
	Return a plugin object. More...
nvinfer1::IPluginV2 *	deserializePlugin (const char *name, const void *serialData, size_t serialLength) override
	Called during deserialization of plugin layer. More...
void	setPluginNamespace (const char *pluginNamespace) override
	Set the namespace of the plugin creator based on the plugin library it belongs to. More...
const char *	getPluginNamespace () const override
	Return the namespace of the plugin creator object. More...
virtual int32_t	getTensorRTVersion () const
	Return the version of the API the plugin creator was compiled with. More...

Private Attributes
std::string	mNamespace

Static Private Attributes
static nvinfer1::PluginFieldCollection	mFC {}
static std::vector< nvinfer1::PluginField >	mPluginAttributes

Constructor & Destructor Documentation

QKVToContextInterleavedPluginCreator()

bert::QKVToContextInterleavedPluginCreator::QKVToContextInterleavedPluginCreator

(

)

Member Function Documentation

getPluginName()

const char * bert::QKVToContextInterleavedPluginCreator::getPluginName ( ) const

overridevirtual

Return the plugin name.

Implements nvinfer1::IPluginCreator.

getPluginVersion()

const char * bert::QKVToContextInterleavedPluginCreator::getPluginVersion ( ) const

overridevirtual

Return the plugin version.

Implements nvinfer1::IPluginCreator.

getFieldNames()

const PluginFieldCollection * bert::QKVToContextInterleavedPluginCreator::getFieldNames ( )

overridevirtual

Return a list of fields that needs to be passed to createPlugin.

See also

PluginFieldCollection

Implements nvinfer1::IPluginCreator.

createPlugin()

IPluginV2 * bert::QKVToContextInterleavedPluginCreator::createPlugin ( const char *  name, const nvinfer1::PluginFieldCollection *  fc  )

overridevirtual

Return a plugin object.

Return nullptr in case of error.

Implements nvinfer1::IPluginCreator.

deserializePlugin()

IPluginV2 * bert::QKVToContextInterleavedPluginCreator::deserializePlugin ( const char *  name, const void *  serialData, size_t  serialLength  )

overridevirtual

Called during deserialization of plugin layer.

Return a plugin object.

Implements nvinfer1::IPluginCreator.

setPluginNamespace()

void bert::QKVToContextInterleavedPluginCreator::setPluginNamespace ( const char *  pluginNamespace )

overridevirtual

Set the namespace of the plugin creator based on the plugin library it belongs to.

This can be set while registering the plugin creator.

See also

IPluginRegistry::registerCreator()

Implements nvinfer1::IPluginCreator.

getPluginNamespace()

const char * bert::QKVToContextInterleavedPluginCreator::getPluginNamespace ( ) const

overridevirtual

Return the namespace of the plugin creator object.

Implements nvinfer1::IPluginCreator.

getTensorRTVersion()

virtual int32_t nvinfer1::IPluginCreator::getTensorRTVersion ( ) const

inlinevirtualinherited

Return the version of the API the plugin creator was compiled with.

Member Data Documentation

mFC

PluginFieldCollection bert::QKVToContextInterleavedPluginCreator::mFC {}

staticprivate

mPluginAttributes

std::vector< PluginField > bert::QKVToContextInterleavedPluginCreator::mPluginAttributes

staticprivate

mNamespace

std::string bert::QKVToContextInterleavedPluginCreator::mNamespace

private

---

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

• qkvToContextInt8InterleavedPlugin.h
• qkvToContextInt8InterleavedPlugin.cpp