Add a second inline_ofg_expansion in xtensor for Ops that wrap OpFromGraph once lowered

Author: ricardoV94

pytensor/xtensor/rewriting/utils.py

@@ -1,6 +1,7 @@
 from pytensor.compile import optdb
-from pytensor.graph.rewriting.basic import NodeRewriter
+from pytensor.graph.rewriting.basic import NodeRewriter, in2out
 from pytensor.graph.rewriting.db import EquilibriumDB, RewriteDatabase
+from pytensor.tensor.rewriting.ofg import inline_ofg_expansion
 
 
 lower_xtensor_db = EquilibriumDB(ignore_newtrees=False)
@@ -14,6 +15,15 @@
     position=0.1,
 )
 
+# Register OFG inline again after lowering xtensor
+optdb.register(
+    "inline_ofg_expansion_xtensor",
+    in2out(inline_ofg_expansion),
+    "fast_run",
+    "fast_compile",
+    position=0.11,
+)
+
 
 def register_lower_xtensor(
     node_rewriter: RewriteDatabase | NodeRewriter | str, *tags: str, **kwargs