diff --git a/tensorflow/compiler/mlir/g3doc/includes/tf_passes.md b/tensorflow/compiler/mlir/g3doc/includes/tf_passes.md index 3d6049ea0e7..77853a14b06 100644 --- a/tensorflow/compiler/mlir/g3doc/includes/tf_passes.md +++ b/tensorflow/compiler/mlir/g3doc/includes/tf_passes.md @@ -75,6 +75,11 @@ func @cluster_oplist(%arg0: tensor, %arg1: tensor) -> tensor { -algorithm : Clustering algorithm type: `use-def` or `union-find` -oplist : Cluster listed ops when they form a single use def-use chain, such that each op's single user is the next op in the list. ``` +### `-prepare-tpu-computation-for-tf-export`: Prepare TPU computation to be legal for export to TensorFlow +Prepares TPU computation module attached to _TPUCompileMlir op for +TensorFlow graph export by making transformation such as replacing or +removing MLIR or XLA specific attributes that are not legal in TensorFlow +graph. ### `-tf-device-cluster-outlining`: Outlines regions of tf_device.cluster operations This pass outlines the body of a `tf_device.cluster` into a function and replaces the `tf_device.cluster` op with an equivalent `tf_device.cluster_func`