[tensorexpr] Allocate intermediate buffers at compile time (#64227)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64227
Test Plan: Imported from OSS
Reviewed By: ZolotukhinM
Differential Revision: D30652220
Pulled By: huiguoo
fbshipit-source-id: cd75005cdfa42751318de7174b44e14a3a01634e
