fix lilstm test in tensorexpr_te (#37913)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37913
Reviewed By: ZolotukhinM
Differential Revision: D21428329
Pulled By: Krovatkin
fbshipit-source-id: eefba49b59dc76a6efaad85f03a4c12b889b60a9