pytorch
24b9b304 - [TensorExpr] Nuke TE shape inference. (#65554)

Commit
4 years ago
[TensorExpr] Nuke TE shape inference. (#65554) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65554 We're relying on JIT based shape inference and not using the TE implementation. Question to the audience: we set `hasBroadcasts_` in that function, but this function was almost never invoked. Do we behave correctly in the presence of rand-calls and broadcasts? Test Plan: Imported from OSS Reviewed By: bertmaher Differential Revision: D31148925 Pulled By: ZolotukhinM fbshipit-source-id: 2898a57e389ea0950163122089d0fec3d92701c4
Author
Mikhail Zolotukhin
Parents
Loading