Searched refs:device_spec (Results 1 – 13 of 13) sorted by relevance
11 device_spec = cfm_usb_devices.HUDDLY_GO12 self.assertEqual(cfm_usb_devices.get_usb_device_spec(device_spec.vid_pid),13 device_spec)
58 device_spec = usb_device_spec.UsbDeviceSpec(63 self.assertTrue(self._usb_device.interfaces_match_spec(device_spec))
55 if not context.context().device_spec.device_type:58 is_remote_device = context.context().device_spec.device_type != "CPU"
139 self.device_spec = _starting_device_spec546 def device_spec(self): member in Context548 return self._thread_local_data.device_spec565 old_device_spec = eager_context.device_spec579 device_spec = pydev.DeviceSpec.from_string(name)586 new_device_spec.merge_from(device_spec)594 eager_context.device_spec = new_device_spec598 eager_context.device_spec = old_device_spec
85 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())89 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())92 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())96 self.assertEqual(ctx.device_name, ctx.device_spec.to_string())
304 device_spec = DeviceSpec.from_string(spec or "")305 _cached_device_specs[spec] = device_spec306 spec = device_spec
4422 for device_spec in self._device_function_stack.peek_objs():4423 if device_spec.function is None:4425 op._set_device(device_spec.function(op))5609 for device_spec in innermost_nonempty_device_stack.peek_objs():5610 if device_spec.function is None:5612 if device_spec.raw_string:5614 context_manager_input = device_spec.raw_string
795 device_spec = tf_device.DeviceSpec(802 device_spec.job = self._job_name803 device_spec.task = task_id804 return device_spec.to_string()
100 NodeBuilder& NodeBuilder::Device(StringPiece device_spec) { in Device() argument101 def_builder_.Device(device_spec); in Device()
103 NodeBuilder& Device(StringPiece device_spec);
209 NodeDefBuilder& NodeDefBuilder::Device(StringPiece device_spec) { in Device() argument210 node_def_.set_device(string(device_spec)); in Device()
91 NodeDefBuilder& Device(StringPiece device_spec);
390 def _dimensions(self, device_spec): argument392 cmd = 'python %s %s' % (device_script, device_spec)