docs: update transformations API docs

google-deepmind · Jun 28, 2024 · b849fc6 · b849fc6
1 parent 5c01f6f
commit b849fc6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/api/transformations.rst b/docs/api/transformations.rst
@@ -174,6 +174,10 @@ Transformations and states
 .. autoclass:: NonNegativeParamsState
     :members:
 
+.. autofunction:: normalize_by_update_norm
+.. autoclass:: NormalizeByUpdateNormState
+    :members:
+
 .. autofunction:: per_example_global_norm_clip
 .. autofunction:: per_example_layer_norm_clip