jax
c5dc980d - [mgpu/pallas_mgpu] Pointwise tanh support

Commit

1 year ago

[mgpu/pallas_mgpu] Pointwise tanh support PiperOrigin-RevId: 700158250

References

#25059 - [mgpu/pallas_mgpu] Pointwise tanh support

#170 - CI: 12/02/24 upstream sync

Author

cperivol

cperivol

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading