Searched refs:colocation_group (Results 1 – 3 of 3) sorted by relevance
455 string colocation_group = ""; in PinArgsAndRets() local456 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()459 << " colo group: " << colocation_group; in PinArgsAndRets()460 while (src_device->empty() && colocation_group.empty() && in PinArgsAndRets()468 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()471 << " colo group: " << colocation_group; in PinArgsAndRets()482 if (!colocation_group.empty()) { in PinArgsAndRets()484 colo_attr.add_s(colocation_group); in PinArgsAndRets()485 std::vector<string> colo_slice = {colocation_group}; in PinArgsAndRets()
331 const Node* node, StringPiece colocation_group);
1006 const Node* node, StringPiece colocation_group) { in ColocateNodeToGroup() argument1007 const Node*& root_node = (*colocation_group_root)[colocation_group]; in ColocateNodeToGroup()1022 << "' with nodes in colocation group '" << colocation_group in ColocateNodeToGroup()