[Pallas TPU] Use vector.broadcast instead of vector.BroadcastOp to fix type check failure #68471
Job | Run time |
---|---|
7m 30s | |
1m 33s | |
5m 40s | |
3m 1s | |
2m 32s | |
1m 19s | |
1m 14s | |
22m 49s |
Job | Run time |
---|---|
7m 30s | |
1m 33s | |
5m 40s | |
3m 1s | |
2m 32s | |
1m 19s | |
1m 14s | |
22m 49s |