tree 0dac903569e3bd99e6fa674eda90293e53996609
parent a00efe55c3790789b967facf10c3f426faa98155
author Jane Xu <janeyx@meta.com> 1669245797 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1669245797 +0000

Rectify `native_batch_norm` schema by splitting it into two legit schemas (#88697)

Using the same repro from the issue (but with BatchNorm2D)

Rectifies native_batch_norm schema by splitting the schema into 2:
1. one will have NON-optional alias-able running_mean and running_var inputs
2. the other will just not have those parameters at all (no_stats variation)

**Calling for name suggestions!**

## test plan
I've added tests in test_functionalization.py as well as an entry in common_method_invocations.py for `native_batch_norm_legit`
CI should pass.

## next steps
Because of bc/fc reasons, we reroute native_batch_norm to call our new schemas ONLY through the python dispatcher, but in 2 weeks or so, we should make `native_batch_norm_legit` the official batch_norm.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/88697
Approved by: https://github.com/albanD