message: OptimizationPass::POST_REWRITE_FOR_EXEC after Grappler optimization in PartitionedCallOp
diff: @@ -166,12 +166,6 @@ class PartitionedCallOp : public AsyncOpKernel {
  OptimizationPassRegistry::Global()->RunGrouping(
  OptimizationPassRegistry::POST_PLACEMENT, optimization_options),
  done);
  - OP_REQUIRES_OK_ASYNC(
  - ctx,
  - OptimizationPassRegistry::Global()->RunGrouping(
  - OptimizationPassRegistry::POST_REWRITE_FOR_EXEC,
  - optimization_options),
  - done);
  Device* cpu_device;
  OP_REQUIRES_OK_ASYNC(
  @@ -184,6 +178,13 @@ class PartitionedCallOp : public AsyncOpKernel {
  device_set, cpu_device, &graph),
  done);
  + OP_REQUIRES_OK_ASYNC(
  + ctx,
  + OptimizationPassRegistry::Global()->RunGrouping(
  + OptimizationPassRegistry::POST_REWRITE_FOR_EXEC,
  + optimization_options),
  + done);
  +
  std::unordered_map<string, std::unique_ptr<Graph>> subgraphs;
  OP_REQUIRES_OK_ASYNC(
  ctx, PartitionHelper(device_set, std::move(graph), &subgraphs),
  