[TensorExpr] LoopNest: add a constructor that takes Stmt instead of list of Tensors. (#45949)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45949
Test Plan: Imported from OSS
Reviewed By: bertmaher
Differential Revision: D24156001
Pulled By: ZolotukhinM
fbshipit-source-id: 6f4f050b04e802e274c42ed64be74c21ba79c29f
Author
Mikhail Zolotukhin