conv3d working

2017-06-29 09:06:29 +00:00
parent ffdb70abee
commit a04c3eaf88
3 changed files with 17 additions and 14 deletions
@@ -14,12 +14,14 @@ Activation::Activation(Network *net, dataDim_t input_dim, tkdnnActivationMode_t
    checkCUDNN( cudnnSetTensor4dDescriptor(srcTensorDesc,
                                        net->tensorFormat,
                                        net->dataType,
-                                        input_dim.n, input_dim.c,
+                                        input_dim.n*input_dim.l, 
+                                        input_dim.c,
                                        input_dim.h, input_dim.w) );
    checkCUDNN( cudnnSetTensor4dDescriptor(dstTensorDesc,
                                            net->tensorFormat,
                                            net->dataType,
-                                            input_dim.n, input_dim.c,
+                                            input_dim.n*input_dim.l, 
+                                            input_dim.c,
                                            input_dim.h, input_dim.w) );
 }