Searched defs:selected_dims (Results 1 – 1 of 1) sorted by relevance
3541 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllGatherShards()3550 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllGatherShardsInternal()3640 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllReduceAlongShardingDims()3650 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllReduceAlongShardingDimsInternal()