diff options
| author |  Tong Shen <endlessroad@google.com> | 2018-09-12 10:03:32 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-12 10:07:48 -0700 |
| commit | 37ddb13ece32500bf87af5d8b8493be1c77781de (patch) | |
| tree | d3b5daf4c29b5402a0db9e3f316665b5c1d1a9b1 /tensorflow/compiler/jit | |
| parent | 26509bf4e202c09da4f0b00d43ebddf87368a0f2 (diff) | |
Roll forward change "Move control flow functionalization as a graph optimization pass, instead of a step in XlaCompiler.".
PiperOrigin-RevId: 212657932
Diffstat (limited to 'tensorflow/compiler/jit')
| -rw-r--r-- | tensorflow/compiler/jit/BUILD | 1 | ||||
| -rw-r--r-- | tensorflow/compiler/jit/jit_compilation_pass_registration.cc | 12 |
2 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/compiler/jit/BUILD b/tensorflow/compiler/jit/BUILD index a989f15a1c..7d5db713f6 100644 --- a/tensorflow/compiler/jit/BUILD +++ b/tensorflow/compiler/jit/BUILD @@ -265,6 +265,7 @@ cc_library( srcs = ["jit_compilation_pass_registration.cc"], deps = [ ":compilation_passes", + "//tensorflow/compiler/tf2xla:functionalize_control_flow_pass_registration", "//tensorflow/core:core_cpu_internal", ], alwayslink = 1, diff --git a/tensorflow/compiler/jit/jit_compilation_pass_registration.cc b/tensorflow/compiler/jit/jit_compilation_pass_registration.cc index c37b6112cc..5dcf754969 100644 --- a/tensorflow/compiler/jit/jit_compilation_pass_registration.cc +++ b/tensorflow/compiler/jit/jit_compilation_pass_registration.cc @@ -21,6 +21,18 @@ limitations under the License. namespace tensorflow { +// PRE_PLACEMENT passes: + +// from +// third_party/tensorflow/compiler/tf2xla/functionalize_control_flow_pass_registration.cc +// FunctionalizeControlFlowPass: 27 +// +// This pass looks at the graph and all associated FunctionDefs, and turns +// traditional control flow structure (Switch/Merge/etc.) into functional +// control flow structure (XlaIf/XlaWhile). Following passes must +// handle those FunctionDef correctly. + +// POST_REWRITE_FOR_EXEC passes: REGISTER_OPTIMIZATION(OptimizationPassRegistry::POST_REWRITE_FOR_EXEC, 10, MarkForCompilationPass); |