This is the complete list of members for bert::SkipLayerNormInterleavedPlugin, including all inherited members.
__attribute__((deprecated)) Dims getOutputDimensions(int32_t | nvinfer1::IPluginV2DynamicExt | protected |
__attribute__((deprecated)) bool isOutputBroadcastAcrossBatch(int32_t | nvinfer1::IPluginV2DynamicExt | protected |
__attribute__((deprecated)) bool canBroadcastInputAcrossBatch(int32_t) const | nvinfer1::IPluginV2DynamicExt | inlineprotected |
__attribute__((deprecated)) bool supportsFormat(DataType | nvinfer1::IPluginV2DynamicExt | protected |
__attribute__((deprecated)) void configurePlugin(const Dims * | nvinfer1::IPluginV2DynamicExt | protected |
attachToContext(cudnnContext *, cublasContext *, IGpuAllocator *) | nvinfer1::IPluginV2Ext | inlinevirtual |
canBroadcastInputAcrossBatch(int32_t inputIndex) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
clone() const override | bert::SkipLayerNormInterleavedPlugin | virtual |
configurePlugin(const nvinfer1::DynamicPluginTensorDesc *in, int nbInputs, const nvinfer1::DynamicPluginTensorDesc *out, int nbOutputs) override | bert::SkipLayerNormInterleavedPlugin | |
nvinfer1::IPluginV2DynamicExt::configurePlugin(const DynamicPluginTensorDesc *in, int32_t nbInputs, const DynamicPluginTensorDesc *out, int32_t nbOutputs)=0 | nvinfer1::IPluginV2DynamicExt | pure virtual |
nvinfer1::IPluginV2Ext::configurePlugin(const Dims *inputDims, int32_t nbInputs, const Dims *outputDims, int32_t nbOutputs, const DataType *inputTypes, const DataType *outputTypes, const bool *inputIsBroadcast, const bool *outputIsBroadcast, PluginFormat floatFormat, int32_t maxBatchSize)=0 | nvinfer1::IPluginV2Ext | pure virtual |
configureWithFormat(const Dims *, int32_t, const Dims *, int32_t, DataType, PluginFormat, int32_t) | nvinfer1::IPluginV2Ext | inlineprotectedvirtual |
destroy() override | bert::SkipLayerNormInterleavedPlugin | virtual |
detachFromContext() | nvinfer1::IPluginV2Ext | inlinevirtual |
enqueue(const nvinfer1::PluginTensorDesc *inputDesc, const nvinfer1::PluginTensorDesc *outputDesc, const void *const *inputs, void *const *outputs, void *workspace, cudaStream_t stream) override | bert::SkipLayerNormInterleavedPlugin | virtual |
nvinfer1::IPluginV2Ext::enqueue(int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream)=0 | nvinfer1::IPluginV2 | pure virtual |
getNbOutputs() const override | bert::SkipLayerNormInterleavedPlugin | virtual |
getOutputDataType(int index, const nvinfer1::DataType *inputTypes, int nbInputs) const override | bert::SkipLayerNormInterleavedPlugin | |
nvinfer1::IPluginV2DynamicExt::getOutputDataType(int32_t index, const nvinfer1::DataType *inputTypes, int32_t nbInputs) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
getOutputDimensions(int outputIndex, const nvinfer1::DimsExprs *inputs, int nbInputs, nvinfer1::IExprBuilder &exprBuilder) override | bert::SkipLayerNormInterleavedPlugin | |
nvinfer1::IPluginV2DynamicExt::getOutputDimensions(int32_t outputIndex, const DimsExprs *inputs, int32_t nbInputs, IExprBuilder &exprBuilder)=0 | nvinfer1::IPluginV2DynamicExt | pure virtual |
nvinfer1::IPluginV2Ext::getOutputDimensions(int32_t index, const Dims *inputs, int32_t nbInputDims)=0 | nvinfer1::IPluginV2 | pure virtual |
getPluginNamespace() const override | bert::SkipLayerNormInterleavedPlugin | virtual |
getPluginType() const override | bert::SkipLayerNormInterleavedPlugin | virtual |
getPluginVersion() const override | bert::SkipLayerNormInterleavedPlugin | virtual |
getSerializationSize() const override | bert::SkipLayerNormInterleavedPlugin | virtual |
getTensorRTVersion() const | nvinfer1::IPluginV2DynamicExt | inlineprotectedvirtual |
getWorkspaceSize(const nvinfer1::PluginTensorDesc *inputs, int nbInputs, const nvinfer1::PluginTensorDesc *outputs, int nbOutputs) const override | bert::SkipLayerNormInterleavedPlugin | |
nvinfer1::IPluginV2DynamicExt::getWorkspaceSize(const PluginTensorDesc *inputs, int32_t nbInputs, const PluginTensorDesc *outputs, int32_t nbOutputs) const =0 | nvinfer1::IPluginV2DynamicExt | pure virtual |
nvinfer1::IPluginV2Ext::getWorkspaceSize(int32_t maxBatchSize) const =0 | nvinfer1::IPluginV2 | pure virtual |
initialize() override | bert::SkipLayerNormInterleavedPlugin | virtual |
int32_t | nvinfer1::IPluginV2DynamicExt | protected |
int32_t | nvinfer1::IPluginV2DynamicExt | protected |
int32_t | nvinfer1::IPluginV2DynamicExt | protected |
isOutputBroadcastAcrossBatch(int32_t outputIndex, const bool *inputIsBroadcasted, int32_t nbInputs) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
kFORMAT_COMBINATION_LIMIT | nvinfer1::IPluginV2DynamicExt | static |
mBeta | bert::SkipLayerNormInterleavedPlugin | private |
mBetaDev | bert::SkipLayerNormInterleavedPlugin | private |
mGamma | bert::SkipLayerNormInterleavedPlugin | private |
mGammaDev | bert::SkipLayerNormInterleavedPlugin | private |
mLayerName | bert::SkipLayerNormInterleavedPlugin | private |
mLd | bert::SkipLayerNormInterleavedPlugin | private |
mNamespace | bert::SkipLayerNormInterleavedPlugin | private |
mParamsOnDevice | bert::SkipLayerNormInterleavedPlugin | private |
mParamWordsize | bert::SkipLayerNormInterleavedPlugin | private |
PluginFormat | nvinfer1::IPluginV2DynamicExt | protected |
PluginFormat | nvinfer1::IPluginV2DynamicExt | protected |
serialize(void *buffer) const override | bert::SkipLayerNormInterleavedPlugin | virtual |
setPluginNamespace(const char *pluginNamespace) override | bert::SkipLayerNormInterleavedPlugin | virtual |
SkipLayerNormInterleavedPlugin(const std::string name, const nvinfer1::Weights &beta, const nvinfer1::Weights &gamma) | bert::SkipLayerNormInterleavedPlugin | |
SkipLayerNormInterleavedPlugin(const std::string name, const void *data, size_t length) | bert::SkipLayerNormInterleavedPlugin | |
SkipLayerNormInterleavedPlugin()=delete | bert::SkipLayerNormInterleavedPlugin | |
supportsFormat(DataType type, PluginFormat format) const =0 | nvinfer1::IPluginV2 | pure virtual |
supportsFormatCombination(int pos, const nvinfer1::PluginTensorDesc *inOut, int nbInputs, int nbOutputs) override | bert::SkipLayerNormInterleavedPlugin | |
nvinfer1::IPluginV2DynamicExt::supportsFormatCombination(int32_t pos, const PluginTensorDesc *inOut, int32_t nbInputs, int32_t nbOutputs)=0 | nvinfer1::IPluginV2DynamicExt | pure virtual |
terminate() override | bert::SkipLayerNormInterleavedPlugin | virtual |
~IPluginV2() | nvinfer1::IPluginV2 | inlineprotectedvirtual |
~IPluginV2DynamicExt() | nvinfer1::IPluginV2DynamicExt | inlineprotectedvirtual |
~IPluginV2Ext() | nvinfer1::IPluginV2Ext | inlinevirtual |