From 0e816bb5308be7c1f8e01cf3167b204cd71bf586 Mon Sep 17 00:00:00 2001
From: ShawnXuan <xiexuanx2@gmail.com>
Date: Wed, 7 Oct 2020 22:52:30 +0800
Subject: [PATCH] add fuse add and model update

---
 LanguageModeling/BERT/util.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/LanguageModeling/BERT/util.py b/LanguageModeling/BERT/util.py
index be68ef3..875a489 100755
--- 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
 
-- 
GitLab