RF TF-layers, fix dim cleanup

rwth-i6 · Oct 12, 2023 · f5b894c · f5b894c
1 parent dcd082b
commit f5b894c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/returnn/tf/frontend_layers/config_entry_points.py b/returnn/tf/frontend_layers/config_entry_points.py
@@ -109,7 +109,7 @@ def _cleanup_net_dict_value(elem):
             # Dim dyn_size_ext might be Tensor[rfl.Layer],
             # but now the TF engine actually wants to have Tensor[tf.Tensor].
             # Reset it now. The TF engine should redefine it again.
-            elem.reset_raw()
+            elem.reset_batch_and_raw()
         return elem
 
     # Do some cleanup.