Reflection Padding native plugin fix ,forgot to added input_dim.c in the plugin creator

2022-01-07 05:41:26 +00:00
parent 7298dcfb2f
commit f189efcbb1
3 changed files with 8 additions and 3 deletions
@@ -94,6 +94,8 @@ namespace nvinfer1{
        static std::vector<PluginField> mPluginAttributes;
        std::string mPluginNamespace;
    };
+
+    REGISTER_TENSORRT_PLUGIN(ReflectionPaddingRTPluginCreator);
 };
 #endif