[TensorExpr] Change `LoopNest::vectorize` to accept `For*` instead of `Stmt*`. (#49696)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/49696
And make it static.
Test Plan: Imported from OSS
Reviewed By: navahgar, nickgg
Differential Revision: D25668695
Pulled By: ZolotukhinM
fbshipit-source-id: 8d7fb507d6f3beca70e868d9e0f4c46247311a99
Author
Mikhail Zolotukhin