Method ConstructNodeDef

tensorflow/core/graph/star_server_graph_partition.cc:210–241

}

Status GraphPartitionerBase::ConstructNodeDef(const Node* node, NodeDef *node_def) {
  *node_def = node->def();
  node_def->set_device(node->assigned_device_name());
  node_def->clear_input();

  int num_inputs = 0;
  std::vector<const Edge*> inputs;
  inputs.resize(node->num_inputs(), nullptr);
  for (const Edge* in_edge : node->in_edges()) {
    if (in_edge->IsControlEdge()) {
      inputs.push_back(in_edge);
    } else {
      num_inputs++;
      inputs[in_edge->dst_input()] = in_edge;
    }
  }

  if (num_inputs != node->num_inputs()) {
    return errors::InvalidArgument("Incomplete graph, missing ",
            (node->num_inputs() - num_inputs),
            " inputs for ", node->def().DebugString());
  }

  for (const Edge* in_edge : inputs) {
    if (in_edge->src()->IsSource()) {
      continue;
    }
    AddInput(node_def, in_edge->src()->name(), in_edge->src_output());
  }

  return Status::OK();
}

Status GraphPartitionerBase::ConstructRecvNodeDef(
    const PartitionOptions &opts,