jax
5ea62154 - Add test for jax2tf conversion of dot general with algorithm.

Commit
1 year ago
Add test for jax2tf conversion of dot general with algorithm. Fixes https://github.com/jax-ml/jax/issues/24236 To be fair, the fix was actually in https://github.com/openxla/xla/pull/18222, but this adds a test to JAX to confirm. PiperOrigin-RevId: 688976685
Author
dfm dfm
Parents
Loading