From e1cd5a109ba07e544c70e2eab684c471bfd88ec6 Mon Sep 17 00:00:00 2001 From: Marcus Chiam Date: Fri, 24 Mar 2023 16:48:24 -0700 Subject: [PATCH] Added utility functions to frozen_dict api --- docs/api_reference/flax.core.frozen_dict.rst | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/docs/api_reference/flax.core.frozen_dict.rst b/docs/api_reference/flax.core.frozen_dict.rst index d9797c58f5..8edac6987d 100644 --- a/docs/api_reference/flax.core.frozen_dict.rst +++ b/docs/api_reference/flax.core.frozen_dict.rst @@ -10,3 +10,7 @@ flax.core.frozen_dict package .. autofunction:: freeze .. autofunction:: unfreeze + +.. autofunction:: copy + +.. autofunction:: pop