pytorch

24c49dbf - [functorch] batch rule : few decomposition ops (#96744)

[functorch] batch rule : few decomposition ops (#96744) Fixes https://github.com/pytorch/pytorch/issues/96741 Pull Request resolved: https://github.com/pytorch/pytorch/pull/96744 Approved by: https://github.com/zou3519