Home
last modified time | relevance | path

Searched refs:colocation_group (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc396 string colocation_group = ""; in PinArgsAndRets() local
397 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
400 << " colo group: " << colocation_group; in PinArgsAndRets()
401 while (src_device->empty() && colocation_group.empty() && in PinArgsAndRets()
409 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
412 << " colo group: " << colocation_group; in PinArgsAndRets()
420 if (colocation_group.empty() && remote_src_device) { in PinArgsAndRets()
421 colocation_group = in PinArgsAndRets()
424 << " colo group: " << colocation_group; in PinArgsAndRets()
435 if (!colocation_group.empty()) { in PinArgsAndRets()
[all …]
Dcolocation_graph.h319 const Node* node, StringPiece colocation_group);
Dcolocation_graph.cc877 const Node* node, StringPiece colocation_group) { in ColocateNodeToGroup() argument
878 const Node*& root_node = (*colocation_group_root)[colocation_group]; in ColocateNodeToGroup()
893 << "' with nodes in colocation group '" << colocation_group in ColocateNodeToGroup()