deconv small fix

0efac253 · zchen0211 · 2d956b82 · 0efac253
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

paddle/operators/conv2d_transpose_cudnn_op.cu paddle/operators/conv2d_transpose_cudnn_op.cu +1 -0

未找到文件。
--- a/paddle/operators/conv2d_transpose_cudnn_op.cu
+++ b/paddle/operators/conv2d_transpose_cudnn_op.cu
@@ -43,6 +43,7 @@ class CudnnConvTransposeOpKernel : public framework::OpKernel<T> {

    std::vector<int> strides = ctx.Attr<std::vector<int>>("strides");
    std::vector<int> paddings = ctx.Attr<std::vector<int>>("paddings");
+    // cudnn v5 does not support dilations
    std::vector<int> dilations = ctx.Attr<std::vector<int>>("dilations");
    int user_workspace_size = ctx.Attr<int>("workspace_size_MB");