Searched defs:match_partition_id (Results 1 – 1 of 1) sorted by relevance
256 HloPredicate match_partition_id = [](const HloInstruction* i) { in MatchReduceScatter() local272 HloPredicate match_partition_id, HloPredicate match_replica_id) { in MatchReduceScatter()