Peter Hawkins 71253ac4c1

Generalize reduce-window padding to support (lo, hi) pairs. (#3728) ...

* Generalize reduce-window padding to support (lo, hi) pairs, as XLA does..

This turns out to simplify the code slightly, too.

* Fix select_and_gather_add batching rule and test.

* Fix documentation text to refer to ReduceWindowWithGeneralPadding.

2020-07-13 09:49:52 -04:00

..

__init__.py

Reimplement argmin/argmax using a single pass variadic reduction. (#3611)

2020-07-01 11:01:22 -04:00

lax_control_flow.py

introduce custom typecheck rules, implement them for cond and scan

2020-06-26 15:31:04 -07:00

lax_fft.py

deflake jax/lax & add to flake8 check (#3310)

2020-06-04 13:50:44 -07:00

lax_parallel.py

fix ppermute test bugs found by @jekbradbury (#3675)

2020-07-07 00:30:08 -07:00

lax.py

Generalize reduce-window padding to support (lo, hi) pairs. (#3728)

2020-07-13 09:49:52 -04:00