Add nnx.graph.flatten to graph.rst autosummary

google · Oct 4, 2024 · 07e06bf · 07e06bf
1 parent 5d31452
commit 07e06bf
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
@@ -14,6 +14,7 @@ graph
 .. autofunction:: iter_graph
 .. autofunction:: clone
 .. autofunction:: call
+.. autofunction:: flatten
 
 .. autoclass:: GraphDef
   :members:
@@ -22,4 +23,4 @@ graph
   :members:
 
 .. autofunction:: update_context
-.. autofunction:: current_update_context
+.. autofunction:: current_update_context