jax
e3fc63ca - [xla:cpu] Support for up to 16 sorted inputs

Commit

1 year ago

[xla:cpu] Support for up to 16 sorted inputs + enable more jax/lax tests for XLA CPU thunks PiperOrigin-RevId: 655249641

References

Author

ezhulenev

Committer

a-googler

Parents