Merge pull request #4255 from 8bitmp3:add-graph-flatten-autosummary

PiperOrigin-RevId: 683680650
google · Oct 8, 2024 · 9f52f66 · 9f52f66
2 parents 2a2187b + 149144e
commit 9f52f66
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docs_nnx/api_reference/flax.nnx/graph.rst b/docs_nnx/api_reference/flax.nnx/graph.rst
@@ -10,6 +10,7 @@ graph
 .. autofunction:: update
 .. autofunction:: pop
 .. autofunction:: state
+.. autofunction:: graph
 .. autofunction:: graphdef
 .. autofunction:: iter_graph
 .. autofunction:: clone
@@ -22,4 +23,4 @@ graph
   :members:
 
 .. autofunction:: update_context
-.. autofunction:: current_update_context
+.. autofunction:: current_update_context