Searched refs:per_dim_ar (Results 1 – 2 of 2) sorted by relevance
240 HloComputation* reduction, bool per_dim_ar);
3612 HloComputation* reduction, bool per_dim_ar) { in AllReduceAlongShardingDimsInternal() argument3613 if (!per_dim_ar) { in AllReduceAlongShardingDimsInternal()