Refactor `infer_shape` method of Ops to find output shapes using `gufunc_signature`#1294

Open

Aarsh-Wankar wants to merge 8 commits intopymc-devs:mainfrom

Aarsh-Wankar:infer_shape_1257

Commits on Mar 21, 2025

Refactor infer_shape methods to utilize _gufunc_to_out_shape for output shape computation
Aarsh-Wankar
committed
Redacted changes for Ops with non-conclusive gufunc_signature
Aarsh-Wankar
committed
fixed ruff format
Aarsh-Wankar
committed
Refactor _gufunc_to_out_shape for giving priority to Constant dimensions along with error handling
Aarsh-Wankar
committed
Remove error handling for inconsistent dimensions in _gufunc_to_out_shape
Aarsh-Wankar
committed
Implement infer_shape method in Op class and remove redundant implementations in subclasses
Aarsh-Wankar
committed
fixed circular import
Aarsh-Wankar
committed
Raise ShapeError instead of NotImplementedError for unimplemented infer_shape in Op class
Aarsh-Wankar
committed