Enable function calls in TEFuser and SpecializeAutogradZero (#43866)
Summary:
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43866
Reviewed By: ezyang
Differential Revision: D23452798
Pulled By: Krovatkin
fbshipit-source-id: 2cff4c905bf1b5d9de56e7869458ffa6fce1f1b5