Searched refs:skipped_src_dims (Results 1 – 1 of 1) sorted by relevance
1230 int64 skipped_src_dims = absl::c_count(src_to_tgt, -1); in TransposeShardingWithCollapsedDims() local1237 perm[src_to_tgt.size() - skipped_src_dims] = i; in TransposeShardingWithCollapsedDims()1238 skipped_src_dims--; in TransposeShardingWithCollapsedDims()