Method ColocateNodes

tensorflow/core/common_runtime/colocation_graph.cc:892–896

Merge the (possibly disjoint) sets containing nodes "x" and "y". Returns OK if the all nodes in the union of these sets can be placed on the same device type. NOTE: If this method returns an error, *this is left in an undefined state.

// NOTE: If this method returns an error, *this is left in an undefined
// state.
Status ColocationGraph::ColocateNodes(const Node& x, const Node& y) {
  int x_root = FindAndUpdateRoot(x.id());
  int y_root = FindAndUpdateRoot(y.id());
  return ColocateNodes(x, x_root, y, y_root);
}

// This overload of ColocateNodes() allows a caller to provide the root node
// ids for the two nodes. For large graphs, this noticeably reduces the