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

Add "Argument classes" section to jax.lax.rst.

Browse Source
This commit is contained in:
Benjamin Chetioui 2020-12-01 17:23:36 +01:00
parent f93ca369a8
commit 9c56277878
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
docs/jax.lax.rst
View File

@ -196,3 +196,14 @@ Linear algebra operators (jax.lax.linalg)
qr
svd
triangular_solve
Argument classes
----------------
.. automodule:: jax.lax
.. autoclass:: ConvDimensionNumbers
.. autoclass:: GatherDimensionNumbers
.. autoclass:: Precision
.. autoclass:: RoundingMethod
.. autoclass:: ScatterDimensionNumbers