Searched refs:colocation_group (Results 1 – 3 of 3) sorted by relevance
332 string colocation_group = ""; in PinArgsAndRets() local333 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()336 << " colo group: " << colocation_group; in PinArgsAndRets()337 while (src_device->empty() && colocation_group.empty() && in PinArgsAndRets()341 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()344 << " colo group: " << colocation_group; in PinArgsAndRets()347 if (!colocation_group.empty()) { in PinArgsAndRets()349 colo_attr.add_s(colocation_group); in PinArgsAndRets()350 std::vector<string> colo_slice = {colocation_group}; in PinArgsAndRets()
192 const Node* node, StringPiece colocation_group);
580 const Node* node, StringPiece colocation_group) { in ColocateNodeToGroup() argument581 const Node*& root_node = (*colocation_group_root)[colocation_group]; in ColocateNodeToGroup()596 << "' with nodes in colocation group '" << colocation_group in ColocateNodeToGroup()