[Pallas TPU] Add lowerings for bf16 jnp.ceil
and jnp.floor
in TPU v6+
#2436
Job | Run time |
---|---|
45s | |
45s |
jnp.ceil
and jnp.floor
in TPU v6+
#2436
Job | Run time |
---|---|
45s | |
45s |