Deconv tensorrt

include/tkDNN/Layer.h

@@ -161,6 +161,7 @@ public:
     virtual dnnType* infer(dataDim_t &dim, dnnType* srcData);
 
     int kernelH, kernelW, strideH, strideW, paddingH, paddingW;
+    bool deConv;
 
 protected:
     cudnnFilterDescriptor_t filterDesc;
@@ -173,8 +174,6 @@ protected:
     void inferCUDNN(dnnType* srcData, bool back = false);
     void*  workSpace;
     size_t ws_sizeInBytes;
-
-    bool deConv;
 };
 
 

include/tkDNN/NetworkRT.h

@@ -75,6 +75,7 @@ public:
 
     nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Layer *l);
     nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Conv2d *l);
+    nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, DeConv2d *l);
     nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Activation *l);
     nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Dense *l);
     nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Pooling *l);