Add special device mapping (#1752)
Summary:
To reuse the inference batch size already set in models' meta.yaml file for different kind of devices like AMD GPUs, this PR add a `SPECIAL_DEVICE_MAPPING` and got the corresponding batch size when executes `determine_batch_size ()`
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1752
Reviewed By: xuzhao9
Differential Revision: D47277610
Pulled By: FindHao
fbshipit-source-id: 188df62eb5089b4729559c288e1ed7401984a2a0