Searched refs:tgt_to_src (Results 1 – 2 of 2) sorted by relevance
1208 absl::Span<int64 const> tgt_to_src) { in TransposeShardingWithCollapsedDims() argument1215 new_src_to_tgt.push_back(tgt_to_src.size()); in TransposeShardingWithCollapsedDims()1216 std::vector<int64> new_tgt_to_src(tgt_to_src.begin(), tgt_to_src.end()); in TransposeShardingWithCollapsedDims()1221 std::vector<int64> tgt_dims_skipping_new(tgt_to_src.size(), -1); in TransposeShardingWithCollapsedDims()1223 for (int64 i = 0; i < tgt_to_src.size(); ++i) { in TransposeShardingWithCollapsedDims()1224 if (tgt_to_src[i] < 0) { in TransposeShardingWithCollapsedDims()1245 std::vector<int64> tgt_tiles(tgt_to_src.size(), 1); in TransposeShardingWithCollapsedDims()1247 if (tgt_to_src[i] >= 0) { in TransposeShardingWithCollapsedDims()
203 absl::Span<int64 const> tgt_to_src);