Searched defs:collectives_creator (Results 1 – 1 of 1) sorted by relevance
4330 const SPMDCollectiveOpsCreator& collectives_creator) { in AllGatherShards()4339 const SPMDCollectiveOpsCreator& collectives_creator, bool per_dim_ag) { in AllGatherShardsInternal()4430 const SPMDCollectiveOpsCreator& collectives_creator, in AllReduceAlongShardingDims()4440 const SPMDCollectiveOpsCreator& collectives_creator, in AllReduceAlongShardingDimsInternal()