diff --git a/tensorflow/compiler/jit/jit_compilation_pass_registration.cc b/tensorflow/compiler/jit/jit_compilation_pass_registration.cc index 4773e8dc562..3907e4b0883 100644 --- a/tensorflow/compiler/jit/jit_compilation_pass_registration.cc +++ b/tensorflow/compiler/jit/jit_compilation_pass_registration.cc @@ -31,10 +31,10 @@ namespace tensorflow { // EncapsulateXlaComputationsPass rewrites computations generated by the // xla.compile() Python code into XlaLaunch nodes. -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 26, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 36, EncapsulateXlaComputationsPass); -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 25, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 35, IntroduceFloatingPointJitterPass); // from diff --git a/tensorflow/compiler/mlir/tensorflow/translate/mlir_import_pass_registration.cc b/tensorflow/compiler/mlir/tensorflow/translate/mlir_import_pass_registration.cc index 7855cf6ac68..210b8caebae 100644 --- a/tensorflow/compiler/mlir/tensorflow/translate/mlir_import_pass_registration.cc +++ b/tensorflow/compiler/mlir/tensorflow/translate/mlir_import_pass_registration.cc @@ -17,7 +17,7 @@ limitations under the License. namespace tensorflow { -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10, MlirImportPass); } // namespace tensorflow diff --git a/tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_pass_registration.cc b/tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_pass_registration.cc index b6c25e3d86b..3359ba08fd1 100644 --- a/tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_pass_registration.cc +++ b/tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_pass_registration.cc @@ -17,7 +17,7 @@ limitations under the License. namespace tensorflow { -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10, MlirRoundtripPass); } // namespace tensorflow diff --git a/tensorflow/compiler/tf2xla/functionalize_control_flow_pass_registration.cc b/tensorflow/compiler/tf2xla/functionalize_control_flow_pass_registration.cc index 0278e9c07ec..26d61399ecc 100644 --- a/tensorflow/compiler/tf2xla/functionalize_control_flow_pass_registration.cc +++ b/tensorflow/compiler/tf2xla/functionalize_control_flow_pass_registration.cc @@ -19,7 +19,7 @@ namespace tensorflow { // This pass is required for some AOT backends and all JIT backends, so this // file exists as a separate lib and will be linked to both AOT and JIT. -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 27, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 37, FunctionalizeControlFlowForXlaPass); } // namespace tensorflow diff --git a/tensorflow/core/common_runtime/accumulate_n_optimizer.cc b/tensorflow/core/common_runtime/accumulate_n_optimizer.cc index 15e58ce0cf8..fd3a020bf70 100644 --- a/tensorflow/core/common_runtime/accumulate_n_optimizer.cc +++ b/tensorflow/core/common_runtime/accumulate_n_optimizer.cc @@ -284,7 +284,7 @@ class AccumulateNV2RemovePass : public GraphOptimizationPass { return Status::OK(); } }; -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10, AccumulateNV2RemovePass); } // namespace diff --git a/tensorflow/core/common_runtime/isolate_placer_inspection_required_ops_pass.cc b/tensorflow/core/common_runtime/isolate_placer_inspection_required_ops_pass.cc index bc1915326d7..54673902e63 100644 --- a/tensorflow/core/common_runtime/isolate_placer_inspection_required_ops_pass.cc +++ b/tensorflow/core/common_runtime/isolate_placer_inspection_required_ops_pass.cc @@ -48,7 +48,7 @@ Status IsolatePlacerInspectionRequiredOpsPass::Run( return status; } -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 25, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 35, IsolatePlacerInspectionRequiredOpsPass); } // namespace tensorflow diff --git a/tensorflow/core/common_runtime/lower_functional_ops.cc b/tensorflow/core/common_runtime/lower_functional_ops.cc index ac79759b9d6..e4fe5f08882 100644 --- a/tensorflow/core/common_runtime/lower_functional_ops.cc +++ b/tensorflow/core/common_runtime/lower_functional_ops.cc @@ -179,7 +179,7 @@ Status LowerFunctionalOpsPass::Run( return Status::OK(); } -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10, LowerFunctionalOpsPass); } // namespace tensorflow diff --git a/tensorflow/core/common_runtime/parallel_concat_optimizer.cc b/tensorflow/core/common_runtime/parallel_concat_optimizer.cc index 486b21e7bf7..2bcd6fb40ef 100644 --- a/tensorflow/core/common_runtime/parallel_concat_optimizer.cc +++ b/tensorflow/core/common_runtime/parallel_concat_optimizer.cc @@ -117,7 +117,7 @@ class ParallelConcatRemovePass : public GraphOptimizationPass { return Status::OK(); } }; -REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, +REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10, ParallelConcatRemovePass); } // namespace