diff --git a/pytensor/tensor/rewriting/basic.py b/pytensor/tensor/rewriting/basic.py index de712b2019..83ee8c2c3b 100644 --- a/pytensor/tensor/rewriting/basic.py +++ b/pytensor/tensor/rewriting/basic.py @@ -949,6 +949,7 @@ def equivalent_up_to_constant_casting(a, b) -> bool: return False +@register_infer_shape @register_useless("shape_unsafe") @register_canonicalize("fast_compile", "shape_unsafe") @register_specialize("shape_unsafe") diff --git a/pytensor/tensor/rewriting/subtensor.py b/pytensor/tensor/rewriting/subtensor.py index 31b8bfd2bd..a1943b55f9 100644 --- a/pytensor/tensor/rewriting/subtensor.py +++ b/pytensor/tensor/rewriting/subtensor.py @@ -526,7 +526,6 @@ def local_subtensor_inc_subtensor(fgraph, node): return -@register_infer_shape @register_useless @register_canonicalize @register_specialize @@ -1263,7 +1262,6 @@ def local_adv_sub1_adv_inc_sub1(fgraph, node): return [r2] -@register_infer_shape @register_specialize @register_stabilize @register_canonicalize