[Pallas TPU] Add lowering for lax.shift_right_arithmetic
#56056
Job | Run time |
---|---|
1m 29s | |
10m 11s | |
6m 8s | |
1m 11s | |
5m 29s | |
1m 21s | |
25m 49s |
lax.shift_right_arithmetic
#56056
Job | Run time |
---|---|
1m 29s | |
10m 11s | |
6m 8s | |
1m 11s | |
5m 29s | |
1m 21s | |
25m 49s |