diff options
author | 2018-09-11 14:17:07 -0700 | |
---|---|---|
committer | 2018-09-11 14:24:33 -0700 | |
commit | 6ebe0abcc6bb3c3b50975cd2550bec2012389673 (patch) | |
tree | 8af5eeeee491e7f5e21f3b629a2a364afe9854c9 | |
parent | 72410969ca8dd7f1be48672c6cb943940edb9f31 (diff) |
Construct placer after the first optimization pass is run.
PiperOrigin-RevId: 212518982
-rw-r--r-- | tensorflow/core/kernels/partitioned_function_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/partitioned_function_ops.cc b/tensorflow/core/kernels/partitioned_function_ops.cc index 7bb403290d..3ab7404ea9 100644 --- a/tensorflow/core/kernels/partitioned_function_ops.cc +++ b/tensorflow/core/kernels/partitioned_function_ops.cc @@ -127,12 +127,12 @@ class PartitionedCallOp : public AsyncOpKernel { optimization_options.graph = &graph; optimization_options.flib_def = overlay_lib; optimization_options.device_set = &device_set; - Placer placer(graph.get(), &device_set); OP_REQUIRES_OK_ASYNC( ctx, OptimizationPassRegistry::Global()->RunGrouping( OptimizationPassRegistry::PRE_PLACEMENT, optimization_options), done); + Placer placer(graph.get(), &device_set); OP_REQUIRES_OK_ASYNC(ctx, placer.Run(), done); OP_REQUIRES_OK_ASYNC( ctx, |