jax
bfc3d3cd - [Pallas TPU] Add lowerings for scalar `sin`, `cos`, `tan` and `tanh`

Commit
1 year ago
[Pallas TPU] Add lowerings for scalar `sin`, `cos`, `tan` and `tanh` This PR is similar to https://github.com/jax-ml/jax/pull/24238 PiperOrigin-RevId: 685842905
Author
Parents
Loading