depth->tensorrt8 patches

2022-01-09 19:24:38 +05:30
parent b75cecb105
commit 4d8f99b441
4 changed files with 46 additions and 18 deletions
@@ -23,6 +23,8 @@
 #include <pluginsRT/ShortcutRT.h>
 #include <pluginsRT/UpsampleRT.h>
 #include <pluginsRT/YoloRT.h>
+#include <pluginsRT/ConstantPaddingRT.h>
+#include <pluginsRT/ReflectionPadding.h>



@@ -94,6 +94,8 @@ namespace nvinfer1{
        static std::vector<PluginField> mPluginAttributes;
        std::string mPluginNamespace;
    };
+
+    REGISTER_TENSORRT_PLUGIN(ReflectionPaddingRTPluginCreator);
 };
 #endif