We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bbab571 commit ac2533aCopy full SHA for ac2533a
tensorflow_recommenders_addons/dynamic_embedding/python/ops/tf_save_restore_patch.py
@@ -175,7 +175,7 @@ class _DynamicEmbeddingSaver(saver.Saver):
175
def _get_dynamic_embedding_save_ops(self):
176
save_ops = tf_utils.ListWrapper([])
177
if not self._var_list:
178
- return save_ops
+ return control_flow_ops.group(save_ops.as_list())
179
180
for var in self._var_list:
181
de_var = None
@@ -202,7 +202,7 @@ def _get_dynamic_embedding_save_ops(self):
202
def _get_dynamic_embedding_restore_ops(self):
203
restore_ops = tf_utils.ListWrapper([])
204
205
- return restore_ops
+ return control_flow_ops.group(restore_ops.as_list())
206
207
208
0 commit comments