Change quantization_local_shard_count to number of slices#3496

RUEI4341 wants to merge 1 commit intoAI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

CIeNET-International:maxtext/user/dora/change_quantization_local_shard_countCIeNET-International/maxtext:maxtext/user/dora/change_quantization_local_shard_countCopy head branch name to clipboard