Searched refs:list_local_devices (Results 1 – 11 of 11) sorted by relevance
31 devices = device_lib.list_local_devices()35 devices = device_lib.list_local_devices(config_pb2.ConfigProto())
25 def list_local_devices(session_config=None): function
50 local_devices = device_lib.list_local_devices()
133 devices = device_lib.list_local_devices()
153 // Devices accessible to the test (e.g. as given by list_local_devices).
48 if all(x.device_type == "CPU" for x in device_lib.list_local_devices()):
4943 if "GPU" in [d.device_type for d in device_lib.list_local_devices()]:
274 for device in device_lib.list_local_devices():
45 devices = device_lib.list_local_devices()
1402 machine_devices = device_lib.list_local_devices(
132 for x in device_lib.list_local_devices():1587 for local_device in device_lib.list_local_devices():