From 0efac253d340b22999407d387a4c2098cb5581c2 Mon Sep 17 00:00:00 2001 From: zchen0211 Date: Wed, 1 Nov 2017 17:16:53 -0700 Subject: [PATCH] deconv small fix --- paddle/operators/conv2d_transpose_cudnn_op.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/operators/conv2d_transpose_cudnn_op.cu b/paddle/operators/conv2d_transpose_cudnn_op.cu index 5a286897e06..61fcfb3bd8f 100644 --- 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 { std::vector strides = ctx.Attr>("strides"); std::vector paddings = ctx.Attr>("paddings"); + // cudnn v5 does not support dilations std::vector dilations = ctx.Attr>("dilations"); int user_workspace_size = ctx.Attr("workspace_size_MB"); -- GitLab