Merge pull request #15733 from qingqing01/release/1.3.3

cherry-pick/Fix batch_norm API for data_layout.

Merge pull request #15733 from qingqing01/release/1.3.3
cherry-pick/Fix batch_norm API for data_layout.
fcdc623f · Tao Luo · GitHub · 6cae0682 · 8cbea0ae · fcdc623f
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +1 -0

未找到文件。
--- 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