Skip to content

Commit

Permalink
Update pass documentation to note that xla_outside_compilation attrib…
Browse files Browse the repository at this point in the history
…ute is deleted

PiperOrigin-RevId: 556823670
  • Loading branch information
changm authored and tensorflower-gardener committed Aug 14, 2023
1 parent a2f494a commit 6d6e73c
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
Original file line number Diff line number Diff line change
Expand Up @@ -1951,7 +1951,8 @@ def OutsideCompiledToHostLaunchPass : Pass<"tf-outside-compiled-to-host-launch",

let description = [{
This pass wraps ops with the same `_xla_outside_compilation`
attribute value in a tf_device.launch op with host device assignment.
attribute value in a tf_device.launch op with host device assignment. The
`_xla_outside_compilation` attribute is deleted from the wrapped ops.

A simple example:

Expand All @@ -1970,7 +1971,7 @@ def OutsideCompiledToHostLaunchPass : Pass<"tf-outside-compiled-to-host-launch",
"tf_device.cluster"() ( {
"tf.A"()
"tf_device.launch"() {
"tf.B"() {_xla_outside_compilation = "cluster1"}
"tf.B"() // Note xla_outside_compilation attribute deleted.
tf_device.return
} {device = "TPU_REPLICATED_HOST_0"} : () -> ()
"tf.C"()
Expand Down

0 comments on commit 6d6e73c

Please sign in to comment.