mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Document all_to_all and ppermute.

Browse Source
This commit is contained in:
Peter Hawkins 2019-10-10 15:19:17 -04:00
parent dd72fd5db7
commit 78132c150d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/jax.lax.rst
View File

@ -143,7 +143,9 @@ Parallelism support is experimental.
.. autosummary::
:toctree: _autosummary
all_to_all
psum
pmax
pmin
ppermute
pswapaxes