Fixed for caffe2/opt/tvm_transformer.cc (#44249)
Summary:
Fixes #https://github.com/pytorch/pytorch/issues/41706
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44249
Reviewed By: gmagogsfm
Differential Revision: D23752331
Pulled By: SplitInfinity
fbshipit-source-id: 1d7297e080bc1e065129259e406af7216f3f0665
