Add compat.v2.compat.v1. It is useful if someone imports from

tensorflow.compat import v2 as tf and then calls tf.compat.v1. PiperOrigin-RevId: 255301700
2019-06-26 17:52:35 -07:00 · 2019-06-26 17:52:35 -07:00 · e6939cfb03
commit e6939cfb03
parent 15f6c30d79
2 changed files with 3 additions and 0 deletions
--- a/tensorflow/api_template.init.py
+++ b/tensorflow/api_template.init.py
@ -149,4 +149,5 @@ if hasattr(_current_module, 'keras'):
  optimizers = keras.optimizers
  initializers = keras.initializers

+compat.v2.compat.v1 = compat.v1
 # pylint: enable=undefined-variable
--- a/tensorflow/api_template_v1.init.py
+++ b/tensorflow/api_template_v1.init.py
@ -141,4 +141,6 @@ try:
    vars()['__all__'].remove('compiler')
 except NameError:
  pass
+
+compat.v2.compat.v1 = compat.v1
 # pylint: enable=undefined-variable