From e8f53d2144d9b50ee19563b9d3c67d0f0ead30c6 Mon Sep 17 00:00:00 2001
From: Haifeng Jin <haifengj@google.com>
Date: Mon, 20 Jul 2020 15:08:44 -0700
Subject: [PATCH] export OperatorNotAllowedInGraphError to public for tf.keras
 to use public API of TensorFlow only. The change won't break any existing
 usages.

PiperOrigin-RevId: 322234301
Change-Id: I7b09e7ce7afa293590f09adbd69d38b59afa6175
---
 tensorflow/python/framework/errors_impl.py           |  6 ++++++
 ...errors.-operator-not-allowed-in-graph-error.pbtxt | 12 ++++++++++++
 .../tools/api/golden/v2/tensorflow.errors.pbtxt      |  4 ++++
 3 files changed, 22 insertions(+)
 create mode 100644 tensorflow/tools/api/golden/v2/tensorflow.errors.-operator-not-allowed-in-graph-error.pbtxt

diff --git a/tensorflow/python/framework/errors_impl.py b/tensorflow/python/framework/errors_impl.py
index 48ed060556d..34daf43372a 100644
--- a/tensorflow/python/framework/errors_impl.py
+++ b/tensorflow/python/framework/errors_impl.py
@@ -48,7 +48,13 @@ class InaccessibleTensorError(ValueError):
   pass
 
 
+@tf_export("errors.OperatorNotAllowedInGraphError", v1=[])
 class OperatorNotAllowedInGraphError(TypeError):
+  """An error is raised for unsupported operator in Graph execution.
+
+  For example, using a `tf.Tensor` as a Python `bool` in Graph execution
+  is not allowed.
+  """
   pass
 
 
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.errors.-operator-not-allowed-in-graph-error.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.errors.-operator-not-allowed-in-graph-error.pbtxt
new file mode 100644
index 00000000000..29478c395a5
--- /dev/null
+++ b/tensorflow/tools/api/golden/v2/tensorflow.errors.-operator-not-allowed-in-graph-error.pbtxt
@@ -0,0 +1,12 @@
+path: "tensorflow.errors.OperatorNotAllowedInGraphError"
+tf_class {
+  is_instance: "<class \'tensorflow.python.framework.errors_impl.OperatorNotAllowedInGraphError\'>"
+  is_instance: "<class \'TypeError\'>"
+  member {
+    name: "args"
+    mtype: "<type \'getset_descriptor\'>"
+  }
+  member_method {
+    name: "__init__"
+  }
+}
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.errors.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.errors.pbtxt
index 0a9ef10ef90..4933f33f30d 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.errors.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.errors.pbtxt
@@ -84,6 +84,10 @@ tf_module {
     name: "OpError"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "OperatorNotAllowedInGraphError"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "OutOfRangeError"
     mtype: "<type \'type\'>"