enable more unit tests in test_nn (#16994)
Summary:
These tests work with ROCm 2.1.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16994
Differential Revision: D14059802
Pulled By: bddppq
fbshipit-source-id: 8e2cbb13196c2e0283d3e02b7f761374bc580751