Searched refs:new_tgt_to_src (Results 1 – 1 of 1) sorted by relevance
1216 std::vector<int64> new_tgt_to_src(tgt_to_src.begin(), tgt_to_src.end()); in TransposeShardingWithCollapsedDims() local1217 new_tgt_to_src.push_back(src_to_tgt.size()); in TransposeShardingWithCollapsedDims()1219 new_tgt_to_src); in TransposeShardingWithCollapsedDims()