On top of #3217, make sure the shape compatibility works before the next Orbax release. #9855
Job | Run time |
---|---|
6s | |
22s | |
48s | |
39s | |
35s | |
3s | |
3m 54s | |
4m 36s | |
6m 17s | |
1m 0s | |
3m 48s | |
4m 41s | |
59s | |
3m 53s | |
4m 42s | |
36m 23s |
Job | Run time |
---|---|
6s | |
22s | |
48s | |
39s | |
35s | |
3s | |
3m 54s | |
4m 36s | |
6m 17s | |
1m 0s | |
3m 48s | |
4m 41s | |
59s | |
3m 53s | |
4m 42s | |
36m 23s |