We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0b49efa commit 081ea64Copy full SHA for 081ea64
docs/source/index.rst
@@ -19,10 +19,13 @@ xla_model
19
.. autofunction:: all_reduce
20
21
.. autofunction:: add_step_closure
22
+.. autofunction:: wait_device_ops
23
24
.. autofunction:: optimizer_step
25
26
.. autofunction:: save
27
+.. autofunction:: rendezvous
28
+.. autofunction:: mesh_reduce
29
30
distributed
31
----------------------------------
@@ -52,3 +55,18 @@ utils
52
55
53
56
.. autoclass:: SampleGenerator
54
57
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