Searched refs:arg_devices (Results 1 – 1 of 1) sorted by relevance
253 avals, arg_devices = unzip2(arg_specs)255 device = _device_from_arg_devices(arg_devices)654 abstract_args, arg_devices = unzip2(arg_specs)667 device = _xla_callable_device(nreps, backend, device, arg_devices)761 def _xla_callable_device(nreps, backend, device, arg_devices): argument769 return _device_from_arg_devices(arg_devices)