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

DOC: add conv_dimension_numbers and ConvGeneralDilatedDimensionNumbers to docs

Browse Source
This commit is contained in:
Jake VanderPlas 2021-10-19 17:18:15 -07:00
parent 349d0d0879
commit 94169b96a8
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

@ -53,6 +53,7 @@ Operators
conj
conv
convert_element_type
conv_dimension_numbers
conv_general_dilated
conv_general_dilated_patches
conv_with_general_padding
@ -208,6 +209,7 @@ Argument classes
.. currentmodule:: jax.lax
.. autoclass:: ConvDimensionNumbers
.. autoclass:: ConvGeneralDilatedDimensionNumbers
.. autoclass:: GatherDimensionNumbers
.. autoclass:: Precision
.. autoclass:: RoundingMethod