Searched refs:non_concat_dims (Results 1 – 1 of 1) sorted by relevance
1551 std::vector<int64> non_concat_dims; in HandleConcatenate() local1552 non_concat_dims.reserve(hlo->shape().rank() - 1); in HandleConcatenate()1555 non_concat_dims.push_back(i); in HandleConcatenate()1558 auto grouped = GroupShardingOnDims(sharding, non_concat_dims); in HandleConcatenate()