blob: d5b12d26df0591fe8052c162cda59d7761d64a33 (
plain)
1
2
3
4
5
6
7
8
9
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 *
|