Searched defs:selected_dims (Results 1 – 1 of 1) sorted by relevance
3501 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllGatherShards()3510 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllGatherShardsInternal()3600 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllReduceAlongShardingDims()3610 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllReduceAlongShardingDimsInternal()