Home
last modified time | relevance | path

Searched refs:pydev (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Ddevice_setter.py17 from tensorflow.python.framework import device as pydev unknown
108 current_device = pydev.DeviceSpec.from_string(op.device or "")
116 ps_device = pydev.DeviceSpec.from_string(self._ps_device)
125 worker_device = pydev.DeviceSpec.from_string(self._worker_device or "")
202 ps_job_name = pydev.DeviceSpec.from_string(ps_device).job
Dsaver.py36 from tensorflow.python.framework import device as pydev unknown
437 pydev.canonical_name(spec.device) for spec in saveable.specs)
/external/tensorflow/tensorflow/python/eager/
Dfunction_context.py22 from tensorflow.python.framework import device as pydev unknown
75 device_functions = (pydev.merge_device(ctx.device_name),)
Dcontext.py38 from tensorflow.python.framework import device as pydev unknown
56 _starting_device_spec = pydev.DeviceSpec.from_string("")
540 context_devices.append(pydev.canonical_name(dev_name))
541 spec = pydev.DeviceSpec.from_string(dev_name)
751 config.service_leader = pydev.canonical_name(service_leader)
1001 elif pydev.is_device_spec(name):
2024 device_spec = pydev.DeviceSpec.from_string(new_device_name)
2029 new_device_spec = pydev.DeviceSpec.from_string(
2033 new_device_spec = pydev.DeviceSpec.from_string("")
Dcore_test.py34 from tensorflow.python.framework import device as pydev unknown
384 device_spec = pydev.DeviceSpec.from_string(device_name)
/external/tensorflow/tensorflow/python/ops/
Dsession_ops.py23 from tensorflow.python.framework import device as pydev unknown
120 return pydev.canonical_name(handle_str.split(";")[-1])
/external/tensorflow/tensorflow/python/framework/
Dimporter.py22 from tensorflow.python.framework import device as pydev unknown
287 coloc_device = pydev.DeviceSpec.from_string(coloc_op.device)
Dops_test.py40 from tensorflow.python.framework import device as pydev unknown
747 pydev.DeviceSpec(
1690 pydev.DeviceSpec(
1754 with g.device(pydev.merge_device("/device:GPU:0")):
1756 with g.device(pydev.merge_device("/job:worker")):
1758 with g.device(pydev.merge_device("/device:CPU:0")):
1760 with g.device(pydev.merge_device("/job:ps")):
1762 with g.device(pydev.merge_device(None)):
1895 with g.device(pydev.merge_device("/job:ps")): # Will be overwritten.
1901 with g.device(pydev.merge_device("/job:ps")):
Dops.py50 from tensorflow.python.framework import device as pydev unknown
130 if isinstance(device_name_or_function, pydev.MergeDevice):
154 self.function = pydev.merge_device(device_name_or_function)
159 self.fast_string_merge = isinstance(self.function, pydev.MergeDevice)
1873 if pydev.is_device_spec(dev_spec):
Dtest_util.py54 from tensorflow.python.framework import device as pydev unknown
3505 device1 = pydev.canonical_name(device1)
3506 device2 = pydev.canonical_name(device2)
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py39 from tensorflow.python.framework import device as pydev unknown
362 job_name = pydev.DeviceSpec.from_string(vars_[0].device).job
520 if isinstance(device, pydev.DeviceSpec):
538 device = pydev.DeviceSpec.from_string(fake_op.device)
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py23 from tensorflow.python.framework import device as pydev unknown
64 parsed_device = pydev.DeviceSpec.from_string(device_string)