Searched refs:_xla_callable_device (Results 1 – 1 of 1) sorted by relevance
667 device = _xla_callable_device(nreps, backend, device, arg_devices)761 def _xla_callable_device(nreps, backend, device, arg_devices): function