jax
5344e7ae - add lax.select broadcasting tests, improve rule

Commit

6 years ago

add lax.select broadcasting tests, improve rule

References

#312 - generalize select batch rule (fixes #311)

Author

mattjj

mattjj

Parents

Loading