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

Fix the display of GDA docs

Browse Source
This commit is contained in:
yashkatariya 2022-03-03 12:19:04 -08:00
parent cf9a900d78
commit f0df7b7348
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
docs/jax.experimental.global_device_array.rst
View File

@ -6,8 +6,6 @@ jax.experimental.global_device_array module
API
---
.. autosummary::
:toctree: _autosummary
GlobalDeviceArray
Shard
.. autoclass:: GlobalDeviceArray
:members:
.. autoclass:: Shard