message: Only add ModuleWrapper when lazy loading is requested or when using TF 1.x (to
  print deprecation warnings).
diff: @@ -242,6 +242,9 @@ __all__ = [_s for _s in dir() if not _s.startswith('_')]
  __all__.extend([_s for _s in _names_with_underscore])
  ''' % underscore_names_str
  + # Add module wrapper if we need to print deprecation messages
  + # or if we use lazy loading.
  + if self._api_version == 1 or self._lazy_loading:
  for dest_module, _ in self._module_imports.items():
  deprecation = 'False'
  has_lite = 'False'
  