[Pallas TPU] Use vector.broadcast instead of vector.BroadcastOp to fix type check failure

[Pallas TPU] Use vector.broadcast instead of vector.BroadcastOp to fix type check failure #381