Test backward pass in unit test.
Summary:
In unit test, run both forward and backward pass.
If the backward pass throws `NotImplementedError`, skip the test since the operator does not support backward pass.
Reviewed By: int3
Differential Revision: D64471087
fbshipit-source-id: c9d0c43544314fc11305f271e8e80f7ba07b2675