Searched defs:num_all_reduces (Results 1 – 1 of 1) sorted by relevance
152 int num_all_reduces = all_reduce_group.size(); in MergeAllReduceGroup() local389 int num_all_reduces = cross_slice_all_reduces.size(); in CombineAllReduceOpsOfSameTypeAndGroupAssignment() local467 int num_all_reduces = all_reduce_group.size(); in CombineAllReduceOpsOfSameTypeAndGroupAssignment() local