Repository navigation

Enable batch support for `windowed_mean|variance` - #1600

Open

nicolaspi wants to merge 14 commits into

tensorflow:mainfrom

nicolaspi:windowed_stats

Open

Enable batch support for `windowed_mean|variance`#1600
nicolaspi wants to merge 14 commits into
tensorflow:mainfrom
nicolaspi:windowed_stats

Commits on Sep 26, 2022
- Enable batch support for `windowed_mean|variance`
  nicolaspi
  committed
  2675210
- Remove unused function
  nicolaspi
  committed
  d48cdfc
- Doc fix
  nicolaspi
  committed
  e020543
- Allow lower rank indices
  nicolaspi
  committed
  c28faa5
- Test against tensors with dynamic shapes
  nicolaspi
  committed
  4446b60
- Fix `take_along_axis` import for jax backend
  nicolaspi
  committed
  56c5c16
- Expose `tensorflow.experimental.numpy` API to numpy and jax backends
  nicolaspi
  committed
  26f4f12
- Rewrite `apply_slice_along_axis` using `np.vectorize`
  nicolaspi
  committed
  169f7f5
- Check for statically known rank
  nicolaspi
  committed
  c90e961
- Documentation
  nicolaspi
  committed
  45dabfa
- Documentation
  nicolaspi
  committed
  df117ee
- Style
  nicolaspi
  committed
  a679e26
- Notation
  nicolaspi
  committed
  385cff7
Commits on Oct 9, 2022
- Remove extra JAX_MODE and NUMPY_MODE setting
  nicolaspi
  committed
  92d7143