[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level#25995
Merged
copybara-service[bot] merged 1 commit intomainfrom test_717691707Jan 21, 2025
+47-2
