Searched defs:selected_dims (Results 1 – 1 of 1) sorted by relevance
4329 int64_t* next_channel_id, absl::Span<const int64_t> selected_dims, in AllGatherShards()4338 int64_t* next_channel_id, absl::Span<const int64_t> selected_dims, in AllGatherShardsInternal()4429 int64_t* next_channel_id, absl::Span<const int64_t> selected_dims, in AllReduceAlongShardingDims()4439 int64_t* next_channel_id, absl::Span<const int64_t> selected_dims, in AllReduceAlongShardingDimsInternal()