Function GetVarOfIdentity

tensorflow/core/graph/star_server_graph_partition.cc:68–82

}

const Node* GetVarOfIdentity(const PartitionOptions& opts,
                             const Node* identity_node) {
  for (const Edge* in : identity_node->in_edges()) {
    const Node* src = in->src();
    if (src->IsVariable() &&
        opts.node_to_loc(src) ==
        opts.node_to_loc(identity_node)) {
        // NOTE(jiankeng.pt): Identity has a control edge,
        // should consider the other edges.
        if (in->IsControlEdge()) continue;
       return src;
    }
  }
  return NULL;
}

void SubGraph::CompleteVariables(const PartitionOptions& opts) {
  set<int> var_id_set;