Searched refs:GroupAllReducesById (Results 1 – 2 of 2) sorted by relevance
129 void GroupAllReducesById(HloModule* module);
235 void ArCrsCombiner::GroupAllReducesById(HloModule* module) { in GroupAllReducesById() function in xla::ArCrsCombiner380 GroupAllReducesById(module); in Run()