Added BatchNorm to NetworkRT (conver_layer)

2022-01-10 22:40:31 +05:30
parent bcf0c4eab3
commit 061bc79a69
2 changed files with 35 additions and 0 deletions
@@ -98,6 +98,7 @@ public:
    nvinfer1::IResizeLayer* convert_layer(nvinfer1::ITensor *input, Upsample *l);
    nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, DeformConv2d *l);
    nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input,Padding *l);
+    nvinfer1::ILayer* convert_layer(nvinfer1::ITensor* input,BatchNorm *l);

 #if NV_TENSORRT_MAJOR > 5 && NV_TENSORRT_MAJOR < 8
    bool serialize(const char *filename);