Skip to content

Commit 081ea64

Browse files
committed
Add missing interfaces to the docs index.
1 parent 0b49efa commit 081ea64

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

docs/source/index.rst

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,13 @@ xla_model
1919
.. autofunction:: all_reduce
2020

2121
.. autofunction:: add_step_closure
22+
.. autofunction:: wait_device_ops
2223

2324
.. autofunction:: optimizer_step
2425

2526
.. autofunction:: save
27+
.. autofunction:: rendezvous
28+
.. autofunction:: mesh_reduce
2629

2730
distributed
2831
----------------------------------
@@ -52,3 +55,18 @@ utils
5255

5356
.. autoclass:: SampleGenerator
5457

58+
.. autoclass:: DataWrapper
59+
60+
.. automodule:: torch_xla.utils.gcsfs
61+
62+
.. autofunction:: open
63+
.. autofunction:: list
64+
.. autofunction:: stat
65+
.. autofunction:: remove
66+
.. autofunction:: rmtree
67+
.. autofunction:: read
68+
.. autofunction:: write
69+
.. autofunction:: generic_read
70+
.. autofunction:: generic_write
71+
72+

0 commit comments

Comments
 (0)