Searched refs:new_src_to_tgt (Results 1 – 1 of 1) sorted by relevance
1214 std::vector<int64> new_src_to_tgt(src_to_tgt.begin(), src_to_tgt.end()); in TransposeShardingWithCollapsedDims() local1215 new_src_to_tgt.push_back(tgt_to_src.size()); in TransposeShardingWithCollapsedDims()1218 return TransposeShardingWithCollapsedDims(source, new_src_to_tgt, in TransposeShardingWithCollapsedDims()