[TensorExpr] Fix imports in tensorexpr benchmarks. (#35830)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35830
Test Plan: Imported from OSS
Differential Revision: D20799464
Pulled By: ZolotukhinM
fbshipit-source-id: 1b5981ad15042f601a9b6eb01a799cdf71200666
Author
Mikhail Zolotukhin
