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

Replace tf.flags with absl flags.

Browse Source 
PiperOrigin-RevId: 339353775
Change-Id: I83c3dba2a2b96f968d2cc5a847eaa17c635c6d2e
This commit is contained in:
Scott Zhu 2020-10-27 16:22:45 -07:00 committed by TensorFlower Gardener
parent b801054029
commit 64300f9afe
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/distribute/tpu_strategy_test_utils.py
View File

@ -19,10 +19,12 @@ from __future__ import division
from __future__ import google_type_annotations
from __future__ import print_function
from absl import flags
from tensorflow.python.distribute import tpu_strategy
from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
from tensorflow.python.eager import remote
from tensorflow.python.platform import flags
# from tensorflow.python.platform import flags
from tensorflow.python.tpu import tpu_strategy_util
FLAGS = flags.FLAGS

3
tensorflow/python/keras/mixed_precision/keras_test.py
View File

@ -19,6 +19,7 @@ from __future__ import print_function
import os
from absl import flags
from absl.testing import parameterized
import numpy as np
@ -53,7 +54,7 @@ from tensorflow.python.keras.saving import save
from tensorflow.python.keras.utils import generic_utils
from tensorflow.python.ops import math_ops
from tensorflow.python.ops import variables
from tensorflow.python.platform import flags
# from tensorflow.python.platform import flags
from tensorflow.python.platform import test
from tensorflow.python.training.experimental import loss_scale as loss_scale_module
from tensorflow.python.training.tracking import util as trackable_utils

3
tensorflow/python/keras/tests/automatic_outside_compilation_test.py
View File

@ -20,6 +20,7 @@ from __future__ import print_function
import os
from absl import flags
import numpy as np
from tensorboard.plugins.histogram import summary_v2 as histogram_summary_v2
@ -45,7 +46,7 @@ from tensorflow.python.keras.layers import pooling as pool_layer_lib
from tensorflow.python.lib.io import file_io
from tensorflow.python.ops import math_ops
from tensorflow.python.ops import summary_ops_v2
from tensorflow.python.platform import flags
# from tensorflow.python.platform import flags
from tensorflow.python.platform import test
from tensorflow.python.summary import summary_iterator
from tensorflow.python.tpu import tpu_strategy_util