Searched refs:per_dim_ag (Results 1 – 2 of 2) sorted by relevance
235 const SPMDCollectiveOpsCreator& collectives_creator, bool per_dim_ag);
3511 const SPMDCollectiveOpsCreator& collectives_creator, bool per_dim_ag) { in AllGatherShardsInternal() argument3525 if (per_dim_ag) { in AllGatherShardsInternal()