fix torch_train build (#30497)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30497
fix torch_train build
Test Plan: buck build //xplat/caffe2:torch_trainAndroid
Reviewed By: dreiss
Differential Revision: D18719662
fbshipit-source-id: a3d06b4068d502dbe29681d9f26906f2b8c7b622
