pytorch

c0674c43 - [vmap] Add max_pool3d batch rule (#99522)

[vmap] Add max_pool3d batch rule (#99522) Also add a helper to integrate `max_pool2d_with_indices` and `max_pool3d_with_indices` Pull Request resolved: https://github.com/pytorch/pytorch/pull/99522 Approved by: https://github.com/zou3519