Home
last modified time | relevance | path

Searched refs:device_lib (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/client/
Ddevice_lib_test.py22 from tensorflow.python.client import device_lib
31 devices = device_lib.list_local_devices()
35 devices = device_lib.list_local_devices(config_pb2.ConfigProto())
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_multi_gpu_test.py24 from tensorflow.python.client import device_lib
50 local_devices = device_lib.list_local_devices()
/external/tensorflow/tensorflow/tools/test/
Dsystem_info_lib.py38 from tensorflow.python.client import device_lib
133 devices = device_lib.list_local_devices()
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt72 [//tensorflow/core:lib_internal_impl] # device_lib
77 [//tensorflow/core/common_runtime:device_factory] # device_lib, tfe, tf_session
82 [//tensorflow/core/common_runtime:session_options] # device_lib, tfe, tf_session
/external/tensorflow/tensorflow/python/kernel_tests/
Ddepthtospace_op_test.py24 from tensorflow.python.client import device_lib
48 if all(x.device_type == "CPU" for x in device_lib.list_local_devices()):
Dcontrol_flow_ops_py_test.py35 from tensorflow.python.client import device_lib
4943 if "GPU" in [d.device_type for d in device_lib.list_local_devices()]:
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50_test.py29 from tensorflow.python.client import device_lib
274 for device in device_lib.list_local_devices():
/external/tensorflow/tensorflow/compiler/tests/
Dstateful_random_ops_test.py27 from tensorflow.python.client import device_lib
45 devices = device_lib.list_local_devices()
/external/tensorflow/tensorflow/python/distribute/v1/
Dall_reduce.py24 from tensorflow.python.framework import device as device_lib unknown
672 d_spec = device_lib.DeviceSpec.from_string(devices[d])
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py29 from tensorflow.python.client import device_lib
1402 machine_devices = device_lib.list_local_devices(
DBUILD45 "//tensorflow/python:device_lib",
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py47 from tensorflow.python.client import device_lib
132 for x in device_lib.list_local_devices():
1587 for local_device in device_lib.list_local_devices():
/external/tensorflow/tensorflow/python/
DBUILD5037 "client/device_lib.py",
5106 name = "device_lib",
5107 srcs = ["client/device_lib.py"],
5317 "//tensorflow/core/common_runtime:session_options", # device_lib, tfe, tf_session
5321 "//tensorflow/core:lib_internal_impl", # device_lib