Searched defs: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 <()151 HloOpcode opcode;152 PrimitiveType accum_type;153 int64 domain_id;154 bool is_cross_shard;155 bool use_global_device_ids;156 std::vector<ReplicaGroup> replica_groups;