Add missing includes (#5434)
* Add missing includes
Currently this is included indirectly through PyTorch includes, but when I remove
the include from PyTorch's headers, the xla build fails.
* [TESTING] Pin PyTorch PR
* Retrigger CI after timeout
* Remove .torch_pin
