Home
last modified time | relevance | path

Searched refs:current_device (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dremote.py207 current_device = device_util.current()
208 if current_device:
209 current_device = device_util.canonicalize(current_device)
210 if current_device and current_device != master_device:
214 (current_device, master_device))
217 if not current_device:
Dcore_test.py65 def current_device(): function
415 self.assertEqual(y.device, current_device())
488 self.assertEndsWith(current_device(), 'CPU:0')
490 self.assertEndsWith(current_device(), 'GPU:0')
494 self.assertEndsWith(current_device(), 'GPU:0')
496 self.assertEndsWith(current_device(), 'CPU:0')
506 self.assertEndsWith(current_device(), 'GPU:0')
507 self.assertEndsWith(current_device(), 'GPU:0')
508 self.assertEndsWith(current_device(), 'CPU:0')
510 self.assertEndsWith(current_device(), 'GPU:0')
/external/tensorflow/tensorflow/python/training/
Ddevice_setter.py114 current_device = pydev.DeviceSpec.from_string(op.device or "")
124 current_job, ps_job = current_device.job, ps_device.job
128 ps_device = ps_device.make_merged_spec(current_device)
132 worker_device = worker_device.make_merged_spec(current_device)
/external/tensorflow/tensorflow/python/framework/
Ddevice.py132 current_device = DeviceSpec.from_string(node_def.device or "")
133 return self._spec.make_merged_spec(current_device)
/external/libcups/cgi-bin/
Dadmin.c31 static int current_device = 0; /* Current device shown */ variable
284 if (current_device == 0 && cgiSupportsMultipart()) in choose_device_cb()
298 cgiSetArray("device_class", current_device, device_class); in choose_device_cb()
299 cgiSetArray("device_id", current_device, device_id); in choose_device_cb()
300 cgiSetArray("device_info", current_device, device_info); in choose_device_cb()
301 cgiSetArray("device_make_and_model", current_device, device_make_and_model); in choose_device_cb()
302 cgiSetArray("device_uri", current_device, device_uri); in choose_device_cb()
303 cgiSetArray("device_location", current_device, device_location); in choose_device_cb()
305 current_device ++; in choose_device_cb()
791 current_device = 0; in do_am_printer()
/external/tensorflow/tensorflow/python/distribute/
Dpacked_distributed_variable.py97 current_device = device_util.canonicalize(device_util.current())
98 return self.get_var_on_device(current_device)
Dvalues.py166 current_device = device_util.canonicalize(device_util.current())
168 if device_util.canonicalize(value.device) == current_device:
702 current_device = device_util.canonicalize(device_util.current())
704 if device_util.canonicalize(value.device) == current_device:
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc744 Device* current_device = ctx->flr()->device(); in IsMultiDevice() local
745 DeviceType current_device_type(current_device->device_type()); in IsMultiDevice()
747 if (!DeviceNameUtils::ParseFullName(current_device->name(), in IsMultiDevice()
750 current_device->name()); in IsMultiDevice()
/external/tensorflow/tensorflow/python/ops/
Dctc_ops.py58 current_device = context.context().device_name
59 if current_device is None:
61 return device.DeviceSpec.from_string(current_device).device_type
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_v2.py1758 current_device = get_device_name()
1759 if current_device is None:
1761 return device.DeviceSpec.from_string(current_device).device_type