implement XLAHooks
and register it to PyTorch when loaded.
#15060
Loading
XLAHooks
and register it to PyTorch when loaded.
#15060