Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dall_reduce_combiner.cc107 struct GroupKey { struct
108 GroupKey(const HloInstruction* hlo, const HloDomainMap& domain_map) in GroupKey() argument
117 bool operator<(const GroupKey& other) const { in operator <() argument
186 std::map<GroupKey, std::vector<HloInstruction*>> opcode_groups; in CreateComputationGroups()
196 opcode_groups[GroupKey(instruction, *domain_map)].push_back(instruction); in CreateComputationGroups()