conv2d ok, but deconv ha different dim with tensorrt

2019-10-30 16:45:46 +01:00
parent f247300469
commit 2594f59d0d
4 changed files with 9 additions and 8 deletions
@@ -166,7 +166,7 @@ public:
 protected:
    cudnnFilterDescriptor_t filterDesc;
    cudnnConvolutionDescriptor_t convDesc;
-    cudnnConvolutionFwdAlgo_t     fwAlgo;
+    cudnnConvolutionFwdAlgo_t     algo;
    cudnnConvolutionBwdDataAlgo_t bwAlgo;
    cudnnTensorDescriptor_t biasTensorDesc;