Searched refs:partial_tile_size (Results 1 – 1 of 1) sorted by relevance
334 int64 partial_tile_size = partial_sharding.tile_assignment().dim(dim); in PartialReplicateReshardCompatibleSharding() local336 if (target_tile_size % partial_tile_size != 0 || in PartialReplicateReshardCompatibleSharding()337 target_tile_size < partial_tile_size) { in PartialReplicateReshardCompatibleSharding()341 if (target_tile_size > partial_tile_size) { in PartialReplicateReshardCompatibleSharding()343 expand_tile_sizes.emplace_back(target_tile_size / partial_tile_size); in PartialReplicateReshardCompatibleSharding()