Signed-off-by: perseusdg <43143075+perseusdg@users.noreply.github.com>

removed all TRT8_DEPRACTED functions
2021-11-19 22:40:52 +05:30
parent 6f936096ae
commit a8c98e3c31
2 changed files with 74 additions and 5 deletions
@@ -96,7 +96,12 @@ public:
    nvinfer1::IPluginV2Layer* convert_layer(nvinfer1::ITensor *input, Upsample *l);
    nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, DeformConv2d *l);

+#if NV_TENSORRT_MAJOR > 5 && NV_TENSORRT_MAJOR < 8
    bool serialize(const char *filename);
+#else
+    bool serialize(const char *filename,nvinfer1::IHostMemory *ptr);
+#endif
+
    bool deserialize(const char *filename);
    void destroy();