Searched defs:collectives_creator (Results 1 – 1 of 1) sorted by relevance
3502 const SPMDCollectiveOpsCreator& collectives_creator) { in AllGatherShards()3511 const SPMDCollectiveOpsCreator& collectives_creator, bool per_dim_ag) { in AllGatherShardsInternal()3601 const SPMDCollectiveOpsCreator& collectives_creator, in AllReduceAlongShardingDims()3611 const SPMDCollectiveOpsCreator& collectives_creator, in AllReduceAlongShardingDimsInternal()