[TensorExpr] Support shape inference in TE for aten::cat. (#42387)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/42387
Test Plan: Imported from OSS
Reviewed By: bertmaher
Differential Revision: D22879281
Pulled By: ZolotukhinM
fbshipit-source-id: 775e46a4cfd91c63196b378ee587cc4434672c89
Author
Mikhail Zolotukhin
