src/transformers/benchmark/benchmark_args.py/PyTorchBenchmarkArguments/device
class PyTorchBenchmarkArguments:    def device(self) -> "torch.device":
        requires_backends(self, ["torch"])
        return self._setup_devices[0]