torch.set\_autocast\_xla\_enabled ================================= .. currentmodule:: torch .. autofunction:: set_autocast_xla_enabled