fix: apply drop_axes squeeze in partial decode path for sharding (#3691) #6389
| Job | Run time |
|---|---|
| 1m 1s | |
| 23m 7s | |
| 15m 32s | |
| 16m 20s | |
| 16m 31s | |
| 17m 44s | |
| 15m 53s | |
| 23m 26s | |
| 17m 15s | |
| 14m 25s | |
| 18m 38s | |
| 12m 59s | |
| 25m 28s | |
| 3s | |
| 3h 38m 22s |
| Job | Run time |
|---|---|
| 1m 1s | |
| 23m 7s | |
| 15m 32s | |
| 16m 20s | |
| 16m 31s | |
| 17m 44s | |
| 15m 53s | |
| 23m 26s | |
| 17m 15s | |
| 14m 25s | |
| 18m 38s | |
| 12m 59s | |
| 25m 28s | |
| 3s | |
| 3h 38m 22s |