Skip to content

[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ #64197

[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+

[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ #64197