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
This commit is contained in:
Anna R 2019-06-26 17:52:35 -07:00 committed by TensorFlower Gardener
parent 15f6c30d79
commit e6939cfb03
2 changed files with 3 additions and 0 deletions

View File

@ -149,4 +149,5 @@ if hasattr(_current_module, 'keras'):
optimizers = keras.optimizers optimizers = keras.optimizers
initializers = keras.initializers initializers = keras.initializers
compat.v2.compat.v1 = compat.v1
# pylint: enable=undefined-variable # pylint: enable=undefined-variable

View File

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