From 4e0e14c7a5b3c0a6f5bee419a5436307f1b2dfdd Mon Sep 17 00:00:00 2001 From: Yusha Arif Date: Mon, 29 Jul 2024 04:46:33 +0000 Subject: [PATCH] fix(backend)(TF)(module.py): minor fixes to the backend Model/Layer classes --- ivy/functional/backends/tensorflow/module.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/ivy/functional/backends/tensorflow/module.py b/ivy/functional/backends/tensorflow/module.py index c1105e4c16240..dbad1e919ab10 100644 --- a/ivy/functional/backends/tensorflow/module.py +++ b/ivy/functional/backends/tensorflow/module.py @@ -233,6 +233,8 @@ def recursive_serialize(d): return {k: recursive_serialize(v) for k, v in d.items()} elif isinstance(d, list): return [recursive_serialize(v) for v in d] + elif isinstance(d, tuple): + return tuple(recursive_serialize(v) for v in d) else: return serialize_obj(d) @@ -254,6 +256,8 @@ def recursive_deserialize(d): return {k: recursive_deserialize(v) for k, v in d.items()} elif isinstance(d, list): return [recursive_deserialize(v) for v in d] + elif isinstance(d, tuple): + return tuple(recursive_serialize(v) for v in d) else: return deserialize_obj(d) @@ -927,6 +931,10 @@ def state_dict(self): def module_dict(self): return self._module_dict + @property + def layers(self): + return self._layers + # Dunder Methods # # ---------------# @store_frame_info @@ -1405,6 +1413,9 @@ def build( self._built = True return + def _lock_state(self): + pass + @tf.autograph.experimental.do_not_convert def register_buffer(self, name: str, value: Union[tf.Tensor, tf.Variable]): self._buffers.update({name: value})