Searched refs:tgt_sharding (Results 1 – 1 of 1) sorted by relevance
1243 auto tgt_sharding = hlo_sharding_util::TransposeSharding(source, perm); in TransposeShardingWithCollapsedDims() local1244 auto reshape_tiles = tgt_sharding.tile_assignment(); in TransposeShardingWithCollapsedDims()