chore: import upstream snapshot with attribution

2026-07-13 13:36:55 +08:00
commit c8a779b1bb
1887 changed files with 3245738 additions and 0 deletions
@@ -0,0 +1,37 @@
+from polygraphy.backend.tf.loader import *
+from polygraphy.backend.tf.runner import *
+
+
+def register_logger_callback():
+    from polygraphy.logger import G_LOGGER
+
+    def set_tf_logging_level(severity_trie):
+        import os
+        from polygraphy import mod
+
+        tf = mod.lazy_import("tensorflow<2.0")
+
+        if not tf.is_installed() or not tf.is_importable():
+            return
+
+        sev = severity_trie.get(G_LOGGER.module_path(os.path.dirname(__file__)))
+        if sev > G_LOGGER.WARNING:
+            tf_sev = tf.compat.v1.logging.ERROR
+            tf_logging_level = "3"
+        elif sev > G_LOGGER.INFO:
+            tf_sev = tf.compat.v1.logging.WARN
+            tf_logging_level = "2"
+        elif sev > G_LOGGER.VERBOSE:
+            tf_sev = tf.compat.v1.logging.INFO
+            tf_logging_level = "1"
+        else:
+            tf_sev = tf.compat.v1.logging.DEBUG
+            tf_logging_level = "0"
+
+        tf.compat.v1.logging.set_verbosity(tf_sev)
+        os.environ["TF_CPP_MIN_LOG_LEVEL"] = tf_logging_level
+
+    G_LOGGER.register_callback(set_tf_logging_level)  # Will be registered when this backend is imported.
+
+
+register_logger_callback()