move header inclusion (#71307)
Summary:
A header is used only in the .cc file and it is included by the public header. This causes errors when I try to include the public header.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/71307
Reviewed By: zou3519
Differential Revision: D33650700
Pulled By: ngimel
fbshipit-source-id: d08dd335208da3aaafe333522d9525976c513151
(cherry picked from commit 94805495a0d30c54f22b0609db177d7ac3e26093)