add rm xla attributes in tf docs (#5950)

apache · Jun 28, 2020 · 17bd06a · 17bd06a
1 parent 2b3d90a
commit 17bd06a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/frontend/tensorflow.rst b/docs/frontend/tensorflow.rst
@@ -79,6 +79,10 @@ The model should be exported with a number of transformations to prepare the mod
                 [
                     "remove_nodes(op=Identity, op=CheckNumerics, op=StopGradient)",
                     "sort_by_execution_order", # sort by execution order after each transform to ensure correct node ordering
+                    "remove_attribute(attribute_name=_XlaSeparateCompiledGradients)",
+                    "remove_attribute(attribute_name=_XlaCompile)",
+                    "remove_attribute(attribute_name=_XlaScope)",
+                    "sort_by_execution_order",
                     "remove_device",
                     "sort_by_execution_order",
                     "fold_batch_norms",