From e6939cfb038bdad6bff68c157a861a28604dcfbe Mon Sep 17 00:00:00 2001
From: Anna R <annarev@google.com>
Date: Wed, 26 Jun 2019 17:52:35 -0700
Subject: [PATCH] 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
---
 tensorflow/api_template.__init__.py    | 1 +
 tensorflow/api_template_v1.__init__.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/tensorflow/api_template.__init__.py b/tensorflow/api_template.__init__.py
index 2d97ea3ca50..b75ec148ae8 100644
--- 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
diff --git a/tensorflow/api_template_v1.__init__.py b/tensorflow/api_template_v1.__init__.py
index 38b635a83e2..4fa92b07051 100644
--- 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