fix: apply drop_axes squeeze in partial decode path for sharding (#3691) #6395
| Job | Run time |
|---|---|
| 56s | |
| 18m 18s | |
| 22m 58s | |
| 13m 9s | |
| 16m 30s | |
| 15m 43s | |
| 18m 14s | |
| 15m 38s | |
| 13m 33s | |
| 18m 3s | |
| 17m 27s | |
| 20m 9s | |
| 23m 52s | |
| 3s | |
| 3h 34m 33s |
| Job | Run time |
|---|---|
| 56s | |
| 18m 18s | |
| 22m 58s | |
| 13m 9s | |
| 16m 30s | |
| 15m 43s | |
| 18m 14s | |
| 15m 38s | |
| 13m 33s | |
| 18m 3s | |
| 17m 27s | |
| 20m 9s | |
| 23m 52s | |
| 3s | |
| 3h 34m 33s |