Searched refs:GroupKey (Results 1 – 1 of 1) sorted by relevance
107 struct GroupKey { struct108 GroupKey(const HloInstruction* hlo, const HloDomainMap& domain_map) in GroupKey() function117 bool operator<(const GroupKey& other) const { in operator <() argument186 std::map<GroupKey, std::vector<HloInstruction*>> opcode_groups; in CreateComputationGroups()196 opcode_groups[GroupKey(instruction, *domain_map)].push_back(instruction); in CreateComputationGroups()