Home
last modified time | relevance | path

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

12

/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)
118 if isinstance(spec, device_spec.DeviceSpecV2):
120 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.py4530 for device_spec in self._device_function_stack.peek_objs():
4531 if device_spec.is_null_merge:
4534 if device_spec.function is None:
4537 device_string = device_spec.string_merge(op)
5742 for device_spec in innermost_nonempty_device_stack.peek_objs():
5743 if device_spec.function is None:
5745 if device_spec.raw_string:
5747 context_manager_input = device_spec.raw_string
/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/lite/experimental/acceleration/configuration/
Dproto_to_flatbuffer.cc203 FlatBufferBuilder* builder, const proto::EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec() argument
205 if (device_spec.device_paths_size() > 0) { in ConvertEdgeTpuDeviceSpec()
207 for (const auto& device_path : device_spec.device_paths()) { in ConvertEdgeTpuDeviceSpec()
217 device_spec.platform_type()), in ConvertEdgeTpuDeviceSpec()
218 device_spec.num_chips(), device_paths_fb, device_spec.chip_family()); in ConvertEdgeTpuDeviceSpec()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.cc171 DeviceNameUtils::ParsedName device_spec = device->parsed_name(); in GetTPUDevices() local
172 device_spec.has_type = true; in GetTPUDevices()
173 device_spec.type = "TPU"; in GetTPUDevices()
175 device_spec.has_id = false; in GetTPUDevices()
177 device_set.FindMatchingDevices(device_spec, &host_tpu_devices); in GetTPUDevices()
/external/chromium-trace/catapult/devil/devil/android/sdk/
Dbundletool.py40 device_spec = {
49 json.dump(device_spec, f)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc126 auto lookup = [&devices](Device device_spec) { in GetTPUDevices() argument
127 device_spec.has_type = true; in GetTPUDevices()
128 device_spec.type = kDeviceTPU; in GetTPUDevices()
130 device_spec.has_id = false; in GetTPUDevices()
133 FindMatchingDevices(devices, device_spec, &host_tpu_devices); in GetTPUDevices()
149 for (const auto& device_spec : llvm::make_range( in GetTPUDevices() local
151 auto host_tpu_devices = lookup(device_spec); in GetTPUDevices()
/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/
Dpywrap_tfe.h417 tensorflow::Safe_PyObjectPtr device_spec; member
431 PyObject* device_spec);
Dcontext.py405 device_spec=_starting_device_spec)
853 def device_spec(self): member in Context
855 return self._thread_local_data.device_spec
857 def _set_device(self, device_name, device_spec): argument
859 self._thread_local_data.device_spec = device_spec
1771 old_device_spec = ctx.device_spec
1786 device_spec = pydev.DeviceSpec.from_string(new_device_name)
1793 new_device_spec = new_device_spec.make_merged_spec(device_spec)
1805 if ctx.device_spec is not new_device_spec:
Dcore_test.py343 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
347 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
350 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
354 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
358 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
365 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
383 device_spec = pydev.DeviceSpec.from_string(device_name)
384 with ctx.device(device_spec):
Dpywrap_tfe_src.cc4272 tensorflow::Safe_PyObjectPtr device_spec; member
4298 PyObject* device_spec) { argument
4315 Py_INCREF(device_spec);
4316 defaults.device_spec.reset(device_spec);
4365 Py_INCREF(defaults->second.device_spec.get());
4366 thread_local_data->device_spec.reset(defaults->second.device_spec.get());
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py393 device_spec = tf_device.DeviceSpec.from_string(
396 if device_spec.job is not None and device_spec.job != "localhost":
398 device_spec.job, device_spec.replica, device_spec.task)
Dtpu_strategy.py49 from tensorflow.python.framework import device_spec
819 job_name = device_spec.DeviceSpecV2.from_string(tpu_devices_flat[0]).job
/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/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/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2.py1321 device_spec = tf_device.DeviceSpec.from_string(device)
1322 if device_spec.device_type != "TPU":
1328 device_ordinal=device_spec.device_index,
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc380 py::handle device_spec) in EagerContextThreadLocalDataWrapper() argument
383 py_eager_context.ptr(), is_eager.ptr(), device_spec.ptr()); in EagerContextThreadLocalDataWrapper()
413 return GetPyObject(&GetData()->device_spec); in get_device_spec()
416 SetPyObject(v, &GetData()->device_spec); in set_device_spec()
/external/tensorflow/tensorflow/python/keras/
Dbackend.py48 from tensorflow.python.framework import device_spec
851 if isinstance(device, device_spec.DeviceSpecV2):
871 return device_spec.DeviceSpecV2.from_string(op.device)
873 return device_spec.DeviceSpecV1.from_string(op.device)

12