We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 692a04e commit 6f0b2b1Copy full SHA for 6f0b2b1
paddle/fluid/operators/conv_transpose_cudnn_op.cu
@@ -72,7 +72,7 @@ class CUDNNConvTransposeOpKernel : public framework::OpKernel<T> {
72
const T* filter_data = filter->data<T>();
73
const std::string data_layout_str = ctx.Attr<std::string>("data_format");
74
const paddle::operators::DataLayout data_layout =
75
- (data_layout_str == "NCHW" ? DataLayout::kNCHW : DataLayout::kNHWC);
+ (data_layout_str != "NHWC" ? DataLayout::kNCHW : DataLayout::kNHWC);
76
77
// if channel_last, transpose to channel_first
78
Tensor input_transpose;
0 commit comments