benchmark
Add test to verify models and inputs are using the expected device
#443

Merged

Commits

Add test to verify models and inputs are using the expected device

aaronenyeshi committed 4 years ago
Fix a bug in tts_angular benchmark not allocating inputs on expected device

aaronenyeshi committed 4 years ago
Minor improvements to check_device

aaronenyeshi committed 4 years ago
Fix small mistake for inputs tensor

aaronenyeshi committed 4 years ago