From a4781244f422de02f5ecd903a5ff834ddf7aae02 Mon Sep 17 00:00:00 2001 From: Micaela Verucchi Date: Wed, 24 Nov 2021 18:16:15 +0100 Subject: [PATCH] Fix merge problem Signed-off-by: Micaela Verucchi --- include/tkDNN/pluginsRT/ActivationLeakyRT.h | 42 +++++++++++++++++++++ 1 file changed, 42 insertions(+) diff --git a/include/tkDNN/pluginsRT/ActivationLeakyRT.h b/include/tkDNN/pluginsRT/ActivationLeakyRT.h index ab97c10..1d98a59 100644 --- a/include/tkDNN/pluginsRT/ActivationLeakyRT.h +++ b/include/tkDNN/pluginsRT/ActivationLeakyRT.h @@ -42,5 +42,47 @@ namespace nvinfer1 { const char *getPluginType() const NOEXCEPT override; + const char *getPluginVersion() const NOEXCEPT override; + + void destroy() NOEXCEPT override; + + const char *getPluginNamespace() const NOEXCEPT override; + + void setPluginNamespace(const char *pluginNamespace) NOEXCEPT override; + + IPluginV2 *clone() const NOEXCEPT override; + + int size; + float slope; + + private: + std::string mPluginNamespace; + }; + + class ActivationLeakyRTPluginCreator : public IPluginCreator { + public: + ActivationLeakyRTPluginCreator(); + + void setPluginNamespace(const char *pluginNamespace) NOEXCEPT override; + + IPluginV2 *deserializePlugin(const char *name, const void *serialData, size_t serialLength) NOEXCEPT override; + + const char *getPluginNamespace() const NOEXCEPT override ; + + IPluginV2 *createPlugin(const char *name, const PluginFieldCollection *fc) NOEXCEPT override ; + + const char *getPluginName() const NOEXCEPT override ; + + const char *getPluginVersion() const NOEXCEPT override ; + + const PluginFieldCollection *getFieldNames() NOEXCEPT override; + + private: + static PluginFieldCollection mFC; + static std::vector mPluginAttributes; + std::string mPluginNamespace; + }; + + REGISTER_TENSORRT_PLUGIN(ActivationLeakyRTPluginCreator); }; \ No newline at end of file