[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 #70499
| Job | Run time |
|---|---|
| 1m 31s | |
| 3m 4s | |
| 11m 4s | |
| 2m 30s | |
| 7m 56s | |
| 1m 18s | |
| 1m 21s | |
| 28m 44s |
| Job | Run time |
|---|---|
| 1m 31s | |
| 3m 4s | |
| 11m 4s | |
| 2m 30s | |
| 7m 56s | |
| 1m 18s | |
| 1m 21s | |
| 28m 44s |