Home
last modified time | relevance | path

Searched refs:device_spec (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Ddevice_spec_test.py23 from tensorflow.python.framework import device_spec
28 TEST_V1_AND_V2 = (("v1", device_spec.DeviceSpecV1),
29 ("v2", device_spec.DeviceSpecV2))
57 d = device_spec.DeviceSpecV1()
79 d = device_spec.DeviceSpecV1(job="foo", replica=12, task=3,
108 d = device_spec.DeviceSpecV1(job="foo", replica=12, task=3,
139 d = device_spec.DeviceSpecV1()
180 d = device_spec.DeviceSpecV1.from_string("/job:foo/replica:0")
183 d.merge_from(device_spec.DeviceSpecV1.from_string("/task:1/device:GPU:2"))
186 d = device_spec.DeviceSpecV1()
[all …]
Ddevice.py24 from tensorflow.python.framework import device_spec
27 DeviceSpec = device_spec.DeviceSpecV2
29 DeviceSpec = device_spec.DeviceSpecV1
47 return isinstance(obj, device_spec.DeviceSpecV2)
116 if isinstance(spec, device_spec.DeviceSpecV2):
118 elif isinstance(spec, device_spec.DeviceSpecV1):
Ddevice_test.py25 from tensorflow.python.framework import device_spec
32 TEST_V1_AND_V2 = (("v1", device_spec.DeviceSpecV1),
33 ("v2", device_spec.DeviceSpecV2))
Dops.py4295 for device_spec in self._device_function_stack.peek_objs():
4296 if device_spec.is_null_merge:
4299 if device_spec.function is None:
4302 device_string = device_spec.string_merge(op)
5540 for device_spec in innermost_nonempty_device_stack.peek_objs():
5541 if device_spec.function is None:
5543 if device_spec.raw_string:
5545 context_manager_input = device_spec.raw_string
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc102 auto lookup = [&devices](Device device_spec) { in GetTPUDevices() argument
103 device_spec.has_type = true; in GetTPUDevices()
104 device_spec.type = kDeviceTPU; in GetTPUDevices()
106 device_spec.has_id = false; in GetTPUDevices()
109 FindMatchingDevices(devices, device_spec, &host_tpu_devices); in GetTPUDevices()
125 for (const auto& device_spec : llvm::make_range( in GetTPUDevices() local
127 auto host_tpu_devices = lookup(device_spec); in GetTPUDevices()
/external/autotest/client/common_lib/cros/cfm/usb/
Dcfm_usb_devices_unittest.py11 device_spec = cfm_usb_devices.HUDDLY_GO
12 self.assertEqual(cfm_usb_devices.get_usb_device_spec(device_spec.vid_pid),
13 device_spec)
Dusb_device_unittest.py58 device_spec = usb_device_spec.UsbDeviceSpec(
63 self.assertTrue(self._usb_device.interfaces_match_spec(device_spec))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-device-spec.pbtxt3 is_instance: "<class \'tensorflow.python.framework.device_spec.DeviceSpecV1\'>"
4 is_instance: "<class \'tensorflow.python.framework.device_spec.DeviceSpecV2\'>"
/external/tensorflow/tensorflow/python/eager/
Dcontext.py184 self.device_spec = _starting_device_spec
758 def device_spec(self): member in Context
760 return self._thread_local_data.device_spec
762 def _set_device(self, device_name, device_spec): argument
764 self._thread_local_data.device_spec = device_spec
1569 old_device_spec = ctx.device_spec
1584 device_spec = pydev.DeviceSpec.from_string(new_device_name)
1591 new_device_spec = new_device_spec.make_merged_spec(device_spec)
1603 if ctx.device_spec is not new_device_spec:
Dcore_test.py341 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
345 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
348 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
352 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
356 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
363 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
381 device_spec = pydev.DeviceSpec.from_string(device_name)
382 with ctx.device(device_spec):
DBUILD150 "//tensorflow/python:device_spec",
/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc100 NodeBuilder& NodeBuilder::Device(StringPiece device_spec) { in Device() argument
101 def_builder_.Device(device_spec); in Device()
Dnode_builder.h103 NodeBuilder& Device(StringPiece device_spec);
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py502 device_spec = tf_device.DeviceSpec.from_string(
505 if device_spec.job is not None and device_spec.job != "localhost":
507 device_spec.job, device_spec.replica, device_spec.task)
Dtpu_strategy.py42 from tensorflow.python.framework import device_spec
249 job_name = device_spec.DeviceSpecV2.from_string(tpu_devices_flat[0]).job
DBUILD538 "//tensorflow/python:device_spec",
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-device-spec.pbtxt3 is_instance: "<class \'tensorflow.python.framework.device_spec.DeviceSpecV2\'>"
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.cc209 NodeDefBuilder& NodeDefBuilder::Device(StringPiece device_spec) { in Device() argument
210 node_def_.set_device(string(device_spec)); in Device()
Dnode_def_builder.h93 NodeDefBuilder& Device(StringPiece device_spec);
/external/autotest/client/site_tests/firmware_TouchMTB/
DtouchbotII_robot_wrapper.py390 def _dimensions(self, device_spec): argument
392 cmd = 'python %s %s' % (device_script, device_spec)
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py593 context.context().device_spec.device_type != "CPU"):
/external/tensorflow/tensorflow/python/
DBUILD1162 ":device_spec",
1184 ":device_spec",
1297 name = "device_spec",
1298 srcs = ["framework/device_spec.py"],
5088 ":device_spec",