Searched refs:tf_device (Results 1 – 9 of 9) sorted by relevance
22 from tensorflow.python.framework import device as tf_device unknown47 d = tf_device.DeviceSpec.from_string(d)51 result = tf_device.DeviceSpec(56 result.merge_from(tf_device.DeviceSpec.from_string(default))
39 from tensorflow.python.framework import device as tf_device unknown70 cpu_device = tf_device.DeviceSpec.from_string(device)71 cpu_device.merge_from(tf_device.DeviceSpec(device_type="CPU", device_index=0))298 d_spec = tf_device.DeviceSpec().parse_from_string(d)346 d_spec = tf_device.DeviceSpec().parse_from_string(d)362 return tf_device.DeviceSpec().parse_from_string(device).device_type == "GPU"397 d_spec = tf_device.DeviceSpec().parse_from_string(d)
35 from tensorflow.python.framework import device as tf_device unknown649 spec = tf_device.DeviceSpec.from_string(device)650 return tf_device.DeviceSpec(
35 from tensorflow.python.framework import device as tf_device unknown374 d_spec = tf_device.DeviceSpec.from_string(d)
31 from tensorflow.python.framework import device as tf_device unknown139 d = tf_device.DeviceSpec.from_string(worker)
26 from tensorflow.python.framework import device as tf_device unknown44 spec = tf_device.DeviceSpec.from_string(tpu_devices[0])
27 from tensorflow.python.framework import device as tf_device unknown135 spec = tf_device.DeviceSpec.from_string(device.name)
97 string tf_device = absl::StrCat("/device:", test_device, ":0"); in TEST() local101 root.WithOpName("fused_batch_norm_tf").WithDevice(tf_device), input, in TEST()
30 from tensorflow.python.framework import device as tf_device unknown795 device_spec = tf_device.DeviceSpec(