[SC] Specify tiling in mosaic kernel so it does not rely on global flag xla_tpu_use_tc_device_shape_on_sc.#4033

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

test_924832145AI-Hypercomputer/maxtext:test_924832145Copy head branch name to clipboard