diff options
Diffstat (limited to 'tensorflow/python/platform/parameterized.py')
-rw-r--r-- | tensorflow/python/platform/parameterized.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/python/platform/parameterized.py b/tensorflow/python/platform/parameterized.py new file mode 100644 index 0000000000..cf01512bc1 --- /dev/null +++ b/tensorflow/python/platform/parameterized.py @@ -0,0 +1,10 @@ +"""Switch between depending on pyglib.gfile or an OSS replacement.""" +# 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_PARAMETERIZED: + from tensorflow.python.platform.default._parameterized import * +else: + from tensorflow.python.platform.google._parameterized import * |