[TF2XLA] Deprecate xla.experimental.compile
PiperOrigin-RevId: 325151668
Change-Id: I1a0ac5d58e8237cf47785034086c7cdc240ba116
diff --git a/RELEASE.md b/RELEASE.md
index b0c785c..62bdc11 100644
--- a/RELEASE.md
+++ b/RELEASE.md
@@ -116,6 +116,8 @@
behavior by adjusting the `l2` parameter.
* <ADD RELEASE NOTES HERE>
* XLA Support:
+ * xla.experimental.compile is deprecated, use
+ `tf.function(experimental_compile=True)` instead
* <ADD RELEASE NOTES HERE>
* Tracing and Debugging:
* <ADD RELEASE NOTES HERE>
diff --git a/tensorflow/python/compiler/xla/xla.py b/tensorflow/python/compiler/xla/xla.py
index 51ad556..59b70f2 100644
--- a/tensorflow/python/compiler/xla/xla.py
+++ b/tensorflow/python/compiler/xla/xla.py
@@ -37,6 +37,7 @@
from tensorflow.python.util import nest
from tensorflow.python.util import tf_inspect
from tensorflow.python.util.compat import collections_abc
+from tensorflow.python.util.deprecation import deprecated
from tensorflow.python.util.tf_export import tf_export
_XLA_COMPILE_ATTR = '_xla_compile_id'
@@ -64,6 +65,10 @@
@tf_export('xla.experimental.compile')
+@deprecated(
+ None, 'xla.experimental.compile is deprecated. Consider using '
+ 'tf.function(experimental_compile=True)',
+ warn_once=True)
def compile(computation, inputs=None): # pylint: disable=redefined-builtin
"""Builds an operator that compiles and runs `computation` with XLA.