Iorest · google-labs-jules · Jan 21, 2026
diff --git a/demos/run_demo.py b/demos/run_demo.py
@@ -8,6 +8,7 @@
 import tensorflow.compat.v1 as tf
 import numpy as np
 from graph_optimizer.utils import load_graph, create_node, create_complex_concat_graph
+from graph_optimizer.utils import create_complex_concat_graph
 from graph_optimizer.runner import OptimizationPipeline
 from graph_optimizer.utils.logger import set_log_level, DEBUG, INFO
 

diff --git a/__init__.py → graph_optimizer/__init__.py b/__init__.py → graph_optimizer/__init__.py
diff --git a/core.py → graph_optimizer/core.py b/core.py → graph_optimizer/core.py
diff --git a/runner.py → graph_optimizer/runner.py b/runner.py → graph_optimizer/runner.py
diff --git a/transforms/__init__.py → graph_optimizer/transforms/__init__.py b/transforms/__init__.py → graph_optimizer/transforms/__init__.py
diff --git a/transforms/combine/__init__.py → ..._optimizer/transforms/combine/__init__.py b/transforms/combine/__init__.py → ..._optimizer/transforms/combine/__init__.py
diff --git a/transforms/combine/concat_combine.py → ...izer/transforms/combine/concat_combine.py b/transforms/combine/concat_combine.py → ...izer/transforms/combine/concat_combine.py
diff --git a/transforms/scalar/__init__.py → ...h_optimizer/transforms/scalar/__init__.py b/transforms/scalar/__init__.py → ...h_optimizer/transforms/scalar/__init__.py
diff --git a/transforms/scalar/algebraic_simplify.py → ...r/transforms/scalar/algebraic_simplify.py b/transforms/scalar/algebraic_simplify.py → ...r/transforms/scalar/algebraic_simplify.py
diff --git a/transforms/scalar/constant_fold.py → ...imizer/transforms/scalar/constant_fold.py b/transforms/scalar/constant_fold.py → ...imizer/transforms/scalar/constant_fold.py
diff --git a/transforms/scalar/cse.py → graph_optimizer/transforms/scalar/cse.py b/transforms/scalar/cse.py → graph_optimizer/transforms/scalar/cse.py
diff --git a/transforms/vectorize/__init__.py → ...ptimizer/transforms/vectorize/__init__.py b/transforms/vectorize/__init__.py → ...ptimizer/transforms/vectorize/__init__.py
diff --git a/transforms/vectorize/pack_vectorize.py → ...er/transforms/vectorize/pack_vectorize.py b/transforms/vectorize/pack_vectorize.py → ...er/transforms/vectorize/pack_vectorize.py
diff --git a/utils/__init__.py → graph_optimizer/utils/__init__.py b/utils/__init__.py → graph_optimizer/utils/__init__.py
diff --git a/utils/generators.py → graph_optimizer/utils/generators.py b/utils/generators.py → graph_optimizer/utils/generators.py
diff --git a/utils/graph_utils.py → graph_optimizer/utils/graph_utils.py b/utils/graph_utils.py → graph_optimizer/utils/graph_utils.py
@@ -6,6 +6,8 @@
 and reusability.
 """
 
+from functools import lru_cache
+
 import os
 import collections
 import numpy as np
@@ -132,6 +134,7 @@ def make_output_shapes_attr(shapes: List[List[int]]) -> attr_value_pb2.AttrValue
 # =======================
 
 
+@lru_cache(maxsize=None)
 def extract_base_name(input_name: str) -> str:
     """
     Extract base node name from input (strip port and control marker).

diff --git a/utils/logger.py → graph_optimizer/utils/logger.py b/utils/logger.py → graph_optimizer/utils/logger.py