diff options
Diffstat (limited to 'tensorflow/python/platform/flags.py')
-rw-r--r-- | tensorflow/python/platform/flags.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/python/platform/flags.py b/tensorflow/python/platform/flags.py new file mode 100644 index 0000000000..d5b12d26df --- /dev/null +++ b/tensorflow/python/platform/flags.py @@ -0,0 +1,10 @@ +"""Switch between depending on pyglib.flags or open-source gflags.""" +# pylint: disable=unused-import +# pylint: disable=g-import-not-at-top +# pylint: disable=wildcard-import +import tensorflow.python.platform +import control_imports +if control_imports.USE_OSS and control_imports.OSS_FLAGS: + from tensorflow.python.platform.default._flags import * +else: + from tensorflow.python.platform.google._flags import * |