args_.push_back(arg);
  }
  Node* dst_node = edge->dst();
  Node* dst_image = node_images.at(dst_node);
  int dst_slot = edge->dst_input();
  args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index;
  graph_->AddEdge(args_[arg_index], 0, dst_image, dst_slot);
  return absl::OkStatus();
}

Status Encapsulator::Subgraph::RecordControlResult(
    const Edge* edge,
    const absl::flat_hash_map<const Node*, Node*>& node_images) {