from tensorflow.python.ops.variables import model_variables from tensorflow.python.ops.variables import moving_average_variables from tensorflow.python.ops.variables import report_uninitialized_variables from tensorflow.python.ops.variables import trainable_variables from tensorflow.python.ops.variables import variables_initializer from tensorflow.python.platform.tf_logging import get_logger # WRAPPER_PLACEHOLDER # Hook external TensorFlow modules. _current_module = _sys.modules[__name__] # Lazy-load estimator. _estimator_module = "tensorflow_estimator.python.estimator.api._v1.estimator" estimator = _LazyLoader("estimator", globals(), _estimator_module) _module_dir = _module_util.get_parent_dir_for_name(_estimator_module) if _module_dir: _current_module.__path__ = [_module_dir] + _current_module.__path__ setattr(_current_module, "estimator", estimator) try: from tensorflow.python.keras.api._v1 import keras _current_module.__path__ = ( [_module_util.get_parent_dir(keras)] + _current_module.__path__) setattr(_current_module, "keras", keras) except ImportError: pass # Explicitly import lazy-loaded modules to support autocompletion. # pylint: disable=g-import-not-at-top if not _six.PY2:
# Import compat before trying to import summary from tensorboard, so that # reexport_tf_summary can get compat from sys.modules. Only needed if using # lazy loading. _current_module.compat.v2 # pylint: disable=pointless-statement # Load tensorflow-io-gcs-filesystem if enabled # pylint: disable=g-import-not-at-top if (_os.getenv('TF_USE_MODULAR_FILESYSTEM', '0') == 'true' or _os.getenv('TF_USE_MODULAR_FILESYSTEM', '0') == '1'): import tensorflow_io_gcs_filesystem as _tensorflow_io_gcs_filesystem # pylint: enable=g-import-not-at-top # Lazy-load estimator. _estimator_module = "tensorflow_estimator.python.estimator.api._v1.estimator" estimator = _LazyLoader("estimator", globals(), _estimator_module) _module_dir = _module_util.get_parent_dir_for_name(_estimator_module) if _module_dir: _current_module.__path__ = [_module_dir] + _current_module.__path__ setattr(_current_module, "estimator", estimator) _keras_module = "keras.api._v1.keras" keras = _LazyLoader("keras", globals(), _keras_module) _module_dir = _module_util.get_parent_dir_for_name(_keras_module) if _module_dir: _current_module.__path__ = [_module_dir] + _current_module.__path__ setattr(_current_module, "keras", keras) # Explicitly import lazy-loaded modules to support autocompletion. # pylint: disable=g-import-not-at-top if not _six.PY2: import typing as _typing