[ROCm] Enable test cases in test_nn.py for ROCm (#52836)
Summary:
Enabling tests in test_nn.py for ROCm because they are passing.
Signed-off-by: Kyle Chen <kylechen@amd.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52836
Reviewed By: H-Huang
Differential Revision: D26725891
Pulled By: mruberry
fbshipit-source-id: 59655a2515ddce92ffc4c55dcf6f28257c05e3c9
