jax
613a0cde - [Pallas] Fix lowering tests for reduction ops

Commit

1 year ago

[Pallas] Fix lowering tests for reduction ops Remove unnecessary skip statements. Also added tests for bf16 types. PiperOrigin-RevId: 707739536

References

Author

ayaka14732

Committer

Google-ML-Automation

Parents