feat: add normalize_tile_assignment function needed for local SPMD #13048

Job	Run time
Check Code Changes / check_files	13s
get-torch-commit	3s
Build PyTorch/XLA / build	34m 23s
TPU tests / tpu-test (test/tpu/run_tests.sh)	24m 58s
torchprime tests / Run torchprime E2E tests	5s
TPU tests / tpu-test (test/tpu/run_expensive_test_1.sh)	17m 29s
CPU tests / test (benchmark_tests)	10m 33s
TPU tests / tpu-test (test/tpu/run_training_tests.sh)	18m 9s
CPU tests / test (python_tests, xla_op1)	17m 44s
CPU tests / test (python_tests, xla_op2)	7m 17s
CPU tests / test (python_tests, xla_op4)	4m 58s
CPU tests / test (python_tests, xla_op3)	9m 57s
CPU tests / test (python_tests, torch_mp_op)	6m 21s
TPU tests / tpu-test (test/tpu/run_expensive_test_2.sh)	23m 43s
TPU tests / tpu-test (test/tpu/run_pallas_test.sh)	5m 53s
CPU tests / test (cpp_tests)	7m 41s
CPU tests / test (python_tests, xla_op5)	6m 40s
Build docs	-1s
	3h 16m 6s

Provide feedback