experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 339345604
Change-Id: I6c1eb07ebbbb01500fe646b8579518e085b45e83
This commit is contained in:
Pavithra Vijay 2020-10-27 15:41:59 -07:00 committed by TensorFlower Gardener
parent f4fc45f577
commit 733b5f94d4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/callbacks_v1.py
View File

@ -167,7 +167,7 @@ class TensorBoard(callbacks.TensorBoard):
def _init_writer(self, model):
"""Sets file writer."""
if context.executing_eagerly():
self.writer = summary_ops_v2.create_file_writer(self.log_dir)
self.writer = summary_ops_v2.create_file_writer_v2(self.log_dir)
if not model.run_eagerly and self.write_graph:
with self.writer.as_default():
summary_ops_v2.graph(K.get_graph(), step=0)

2
tensorflow/python/keras/tests/summary_ops_test.py
View File

@ -41,7 +41,7 @@ class SummaryOpsTest(test.TestCase):
def keras_model(self, *args, **kwargs):
logdir = self.get_temp_dir()
writer = summary_ops.create_file_writer(logdir)
writer = summary_ops.create_file_writer_v2(logdir)
with writer.as_default():
summary_ops.keras_model(*args, **kwargs)
writer.close()