Searched refs:device_lib (Results 1 – 13 of 13) sorted by relevance
22 from tensorflow.python.client import device_lib31 devices = device_lib.list_local_devices()35 devices = device_lib.list_local_devices(config_pb2.ConfigProto())
25 from tensorflow.python.client import device_lib50 local_devices = device_lib.list_local_devices()
35 from tensorflow.python.client import device_lib129 devices = device_lib.list_local_devices()
26 from tensorflow.python.client import device_lib40 devices = device_lib.list_local_devices()
27 from tensorflow.python.client import device_lib194 for device in device_lib.list_local_devices():
31 from tensorflow.python.client import device_lib183 for device in device_lib.list_local_devices():
28 from tensorflow.python.client import device_lib208 for device in device_lib.list_local_devices():
24 from tensorflow.python.framework import device as device_lib unknown666 d_spec = device_lib.DeviceSpec.from_string(devices[d])
25 from tensorflow.python.client import device_lib1096 machine_devices = device_lib.list_local_devices(session_config=session_config)
57 "//tensorflow/python:device_lib",
54 from tensorflow.python.client import device_lib101 for x in device_lib.list_local_devices():1296 for local_device in device_lib.list_local_devices():
4003 "client/device_lib.py",4226 name = "device_lib",4227 srcs = ["client/device_lib.py"],4325 "client/device_lib.i",
32 from tensorflow.python.client import device_lib4462 if "GPU" in [d.device_type for d in device_lib.list_local_devices()]: