Merge pull request #15657 from qingqing01/fix_bn_api

Fix batch_norm API for data_layout.
6 years ago · 25ec6a2cfd
parent c47e258ea4 01d9bf9264
commit 25ec6a2cfd
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -2930,6 +2930,7 @@ def batch_norm(input,
            "momentum": momentum,
            "epsilon": epsilon,
            "is_test": is_test,
+            "data_layout": data_layout,
            "use_mkldnn": False,
            "fuse_with_relu": fuse_with_relu,
            "use_global_stats": use_global_stats