Fwd-Bwd correctness tests for TBEs, kernels (#2152)

Summary:
Pull Request resolved: #2152

Adding more tests for kernels coverage, testing inductor compilation and forward-backward numerical correctness.

Reviewed By: TroyGarden

Differential Revision: D58869080

Author: Ivan Kobzarev