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

Remove the usage of graph_and_eager_modes in keras/distribute.

Browse Source 
PiperOrigin-RevId: 341092121
Change-Id: I98f81869c63b77ff1e8c9923980d77f48730e5f5
This commit is contained in:
Scott Zhu 2020-11-06 11:57:52 -08:00 committed by TensorFlower Gardener
parent 70ed67389b
commit 3d28521352
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/keras/distribute/step_fn_test.py
View File

@ -37,8 +37,7 @@ class SingleLossStepTest(test.TestCase, parameterized.TestCase):
@ds_combinations.generate(
combinations.times(
optimizer_combinations.distributions_and_v1_optimizers(),
combinations.combine(
mode=strategy_combinations.graph_and_eager_modes),
combinations.combine(mode=["eager", "graph"]),
combinations.combine(is_tpu=[False])) + combinations.combine(
distribution=[strategy_combinations.tpu_strategy],
optimizer_fn=optimizer_combinations.optimizers_v1,