jax
2d74037d - Change select_and_scatter_add's batching rule to avoid unrolling.

Commit

5 years ago

Change select_and_scatter_add's batching rule to avoid unrolling. Add a test for select_and_scatter_add batching.

References

Author

hawkinsp

Committer

hawkinsp

Parents