Update deprecated import paths. (#67250)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67250
Test Plan: Run tests manually
Reviewed By: NicolasHug
Differential Revision: D31921656
fbshipit-source-id: e2cba7bc7d4a8c7f836bc32f1b8b11a37494a4e2
