Home
last modified time | relevance | path

Searched refs:DeviceSpecV1 (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Ddevice_spec_test.py28 TEST_V1_AND_V2 = (("v1", device_spec.DeviceSpecV1),
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()
187 d.merge_from(device_spec.DeviceSpecV1.from_string("/task:1/cpu:0"))
190 d.merge_from(device_spec.DeviceSpecV1.from_string("/job:boo/device:GPU:0"))
[all …]
Ddevice.py29 DeviceSpec = device_spec.DeviceSpecV1
120 elif isinstance(spec, device_spec.DeviceSpecV1):
Ddevice_test.py32 TEST_V1_AND_V2 = (("v1", device_spec.DeviceSpecV1),
Ddevice_spec.py402 class DeviceSpecV1(DeviceSpecV2): class
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-device-spec.pbtxt3 is_instance: "<class \'tensorflow.python.framework.device_spec.DeviceSpecV1\'>"
/external/tensorflow/tensorflow/python/keras/
Dbackend.py873 return device_spec.DeviceSpecV1.from_string(op.device)