Add test to verify models and inputs are using the expected device (#443)
* Add test to verify models and inputs are using the expected device
The device attribute needs to be set in three locations: the BenchmarkModel,
the nn.Module model, and the model inputs. This test will verify that
the tensors and inputs are on the expected device set in BenchmarkModel.
Also, fix a few benchmark bugs where the device is incorrectly set.
* Fix a bug in tts_angular benchmark not allocating inputs on expected device
* Minor improvements to check_device
Change Exception to RuntimeError.
Update check inputs to a recursive implementation.
* Fix small mistake for inputs tensor