Searched refs:requested_devices (Results 1 – 2 of 2) sorted by relevance
72 def GetDevices(requested_devices, denylist_file): argument81 elif requested_devices:82 requested = set(requested_devices)
1391 requested_devices = set(device_util.canonicalize(d) for d in devices)1406 if device_util.canonicalize(d.name) in requested_devices:1409 if len(using_devices) != len(requested_devices):1412 "to TensorFlow: %s", ",".join(list(requested_devices - using_devices)))1414 if any("gpu" not in d.lower() for d in requested_devices):