infer_shape does not survive through all exceptions anymore, and now uses ShapeError instead of NotImplementedError to catch situations where a shape cannot be computed