add fuse add and model update

0e816bb5 · ShawnXuan · 2f68c17c · 0e816bb5
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

LanguageModeling/BERT/util.py LanguageModeling/BERT/util.py +2 -0

未找到文件。
--- a/LanguageModeling/BERT/util.py
+++ b/LanguageModeling/BERT/util.py
@@ -175,5 +175,7 @@ def GetFunctionConfig(args):
    config.enable_auto_mixed_precision(args.use_fp16)
    if args.use_xla:
        config.use_xla_jit(True)
+    config.enable_fuse_add_to_output(True)
+    config.enable_fuse_model_update_ops(True)
    return config