Turn on batched_grad testing for NewModuleTest (#50740)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/50740
This PR adds a `check_batched_grad=True` option to
NewModuleTest-generated NN tests.
Test Plan: - run tests (`pytest test/test_nn.py -v -rf`)
Reviewed By: ejguan
Differential Revision: D25997679
Pulled By: zou3519
fbshipit-source-id: b75e73d7e86fd3af9bad6efed7127b36551587b3
