Fix mistake of batch norm op (#21237)

* fix_bn * revert unittest,test=develop
6 years ago · b6ce4f8b2f
parent 41d13209d7
commit b6ce4f8b2f
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/batch_norm_op.cu
+++ b/paddle/fluid/operators/batch_norm_op.cu
@ -418,7 +418,7 @@ class BatchNormGradKernel<platform::CUDADeviceContext, T>
          }
        } else {
          if (d_x) {
-            BNBackwardData<T, block, framework::DataLayout::kNCHW><<<
+            BNBackwardData<T, block, framework::DataLayout::kNHWC><<<
                grid2, block, 0, dev_ctx.stream()>>>(
                d_y->data<T>(), scale->data<BatchNormParamType<T>>(),
                saved_mean_data, x->data<T>(), saved_var_data, C, N, H * W * D,