Searched refs:ColocateNodes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.h | 338 Status ColocateNodes(const Node& x, const Node& y);
344 Status ColocateNodes(const Node& x, int x_root, const Node& y, int y_root);
|
| D | colocation_graph.cc | 702 Status status = ColocateNodes(*src, src_root_id, *dst, dst_root_id); in ColocateResourceOrRefEdge()
983 TF_RETURN_IF_ERROR(ColocateNodes(*nodes[0].first, *nodes[i].first)); in ApplyIOColocationGroups()
1015 Status s = ColocateNodes(*node, *root_node); in ColocateNodeToGroup()
1038 Status ColocationGraph::ColocateNodes(const Node& x, const Node& y) { in ColocateNodes() function in tensorflow::ColocationGraph
1041 return ColocateNodes(x, x_root, y, y_root); in ColocateNodes()
1047 Status ColocationGraph::ColocateNodes(const Node& x, int x_root, const Node& y, in ColocateNodes() function in tensorflow::ColocationGraph
|