Update ThroughputBenchmark to reflect new script::Module API (no (#22291)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/22291
There was a race between landing the benchmark diff and
https://github.com/pytorch/pytorch/pull/21934 from zdevito. This PR
should fix the issue.
Reviewed By: zdevito
Differential Revision: D16023640
fbshipit-source-id: 931714352e656f045f9ef3cd17422db51b168384