Searched refs:transpose_sharding (Results 1 – 1 of 1) sorted by relevance
380 auto transpose_sharding = hlo_sharding_util::TransposeSharding( in PartialReplicateReshardCompatibleSharding() local387 auto transpose_tile_assignment = transpose_sharding.tile_assignment(); in PartialReplicateReshardCompatibleSharding()