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

Commit
5 years ago
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.
Author
Parents
Loading