RF TF runtime checks, meta check

rwth-i6 · Oct 12, 2023 · a6e6d24 · a6e6d24
1 parent 61b75cc
commit a6e6d24
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/returnn/tf/frontend_low_level/_backend.py b/returnn/tf/frontend_low_level/_backend.py
@@ -346,7 +346,9 @@ def runtime_sanity_checks(tensor: _TT) -> tf.Operation:
                             data + ["-> invalid shape[%i] or max(dyn_size[%i])" % (i, i)],
                         )
                     ]
-                    checks += [dyn_size_ext.get_runtime_sanity_check_op()]
+                    dyn_size_ext_sanity_checks_op = dyn_size_ext.get_runtime_sanity_check_op()
+                    assert dyn_size_ext_sanity_checks_op is not None, f"{dyn_size_ext} {dyn_size_ext.raw_tensor}?"
+                    checks += [dyn_size_ext_sanity_checks_op]
             return tf.group(*checks)
 
     @staticmethod