[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 #70500
| Job | Run time |
|---|---|
| 4m 33s | |
| 3m 1s | |
| 11m 49s | |
| 1m 32s | |
| 8m 36s | |
| 1m 23s | |
| 1m 16s | |
| 32m 10s |
| Job | Run time |
|---|---|
| 4m 33s | |
| 3m 1s | |
| 11m 49s | |
| 1m 32s | |
| 8m 36s | |
| 1m 23s | |
| 1m 16s | |
| 32m 10s |