Causal Conv1d Plugin

class CausalConv1dPlugin : public nvinfer1::IPluginV2DynamicExt

Public Functions

CausalConv1dPlugin(

std::string const &name,

int32_t stride,

int32_t padding,

int32_t dilation,

int32_t groups

)

CausalConv1dPlugin(

std::string const &name,

void const *data,

size_t length

)

CausalConv1dPlugin() = delete

CausalConv1dPlugin(CausalConv1dPlugin const&) = delete

~CausalConv1dPlugin() override

nvinfer1::IPluginV2DynamicExt *clone() const noexcept override

int32_t getNbOutputs() const noexcept override

nvinfer1::DataType getOutputDataType(

int32_t index,

nvinfer1::DataType const *inputTypes,

int32_t nbInputs

) const noexcept override

nvinfer1::DimsExprs getOutputDimensions(

int32_t outputIndex,

nvinfer1::DimsExprs const *inputs,

int32_t nbInputs,

nvinfer1::IExprBuilder &exprBuilder

) noexcept override

bool supportsFormatCombination(

int32_t pos,

nvinfer1::PluginTensorDesc const *inOut,

int32_t nbInputs,

int32_t nbOutputs

) noexcept override

void configurePlugin(

nvinfer1::DynamicPluginTensorDesc const *in,

int32_t nbInputs,

nvinfer1::DynamicPluginTensorDesc const *out,

int32_t nbOutputs

) noexcept override

size_t getWorkspaceSize(

nvinfer1::PluginTensorDesc const *inputs,

int32_t nbInputs,

nvinfer1::PluginTensorDesc const *outputs,

int32_t nbOutputs

) const noexcept override

int32_t enqueue(

nvinfer1::PluginTensorDesc const *inputDesc,

nvinfer1::PluginTensorDesc const *outputDesc,

void const *const *inputs,

void *const *outputs,

void *workspace,

cudaStream_t stream

) noexcept override

size_t getSerializationSize() const noexcept override

void serialize(void *buffer) const noexcept override

char const *getPluginType() const noexcept override

char const *getPluginNamespace() const noexcept override

void setPluginNamespace(

char const *pluginNamespace

) noexcept override

char const *getPluginVersion() const noexcept override

int32_t initialize() noexcept override

void terminate() noexcept override

void destroy() noexcept override

class CausalConv1dPluginCreator : public nvinfer1::IPluginCreator

Public Functions

CausalConv1dPluginCreator()

~CausalConv1dPluginCreator() override = default

char const *getPluginName() const noexcept override

char const *getPluginVersion() const noexcept override

nvinfer1::PluginFieldCollection const *getFieldNames(

) noexcept override

char const *getPluginNamespace() const noexcept override

void setPluginNamespace(

char const *pluginNamespace

) noexcept override

nvinfer1::IPluginV2 *createPlugin(

char const *name,

nvinfer1::PluginFieldCollection const *fc

) noexcept override

nvinfer1::IPluginV2 *deserializePlugin(

char const *name,

void const *serialData,

size_t serialLength

) noexcept override