fix lower_to_trt api (#822)
Summary:
as titled.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/822
Reviewed By: xuzhao9
Differential Revision: D35117050
Pulled By: frank-wei
fbshipit-source-id: 6e043be79b62005d43b7f489d425ced014674e82
