RF is_executing_eagerly

rwth-i6 · Mar 11, 2024 · 870826b · 870826b
1 parent 7bbfab0
commit 870826b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/returnn/frontend/_backend.py b/returnn/frontend/_backend.py
@@ -1329,6 +1329,13 @@ def get_selected_backend() -> Optional[str]:
     return global_backend.__class__.name
 
 
+def is_executing_eagerly() -> bool:
+    """
+    :return: whether the current selected backend is executing eagerly
+    """
+    return global_backend.executing_eagerly()
+
+
 def select_backend_tf():
     """
     Selects the RETURNN layers backend (based on TF).

diff --git a/returnn/frontend/backend.py b/returnn/frontend/backend.py
@@ -7,13 +7,20 @@
 from . import _backend
 
 # And some functions from the internal backend API.
-from ._backend import select_backend, get_selected_backend, select_backend_torch, select_backend_returnn_layers_tf
+from ._backend import (
+    select_backend,
+    get_selected_backend,
+    select_backend_torch,
+    select_backend_returnn_layers_tf,
+    is_executing_eagerly,
+)
 
 __all__ = [
     "select_backend",
     "get_selected_backend",
     "select_backend_torch",
     "select_backend_returnn_layers_tf",
+    "is_executing_eagerly",
     "is_backend_raw_tensor_dim_tag_independent",
 ]