xla

ba6a9f5b - Move torch import closer to the place it uses (#6412)

Move torch import closer to the place it uses (#6412) Summary: Move torch import closer to the place it uses such that we don't have to have torch installed to do python setup.py clean. Test Plan: Tested locally.