Searched refs:device_string (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op_test.cc | 54 string device_string = "CPU"; in PersistentMemoryTrackingTest() local
57 device_string = "GPU"; in PersistentMemoryTrackingTest()
61 device_string, {}, "/job:worker/replica:0/task:0")); in PersistentMemoryTrackingTest()
|
| /external/libbrillo/brillo/blkdev_utils/ |
| D | loop_device.cc | 64 std::string device_string; in GetDeviceNumber() local
69 if (!base::ReadFileToString(device_file, &device_string)) in GetDeviceNumber()
73 device_string, ":", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY); in GetDeviceNumber()
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 54 def set_cpu0(device_string): argument
65 parsed_device = pydev.DeviceSpec.from_string(device_string)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 1098 def check_device(path, device_string): argument
1099 spec = tf_device.DeviceSpec.from_string(device_string)
1108 path, device_string))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4537 device_string = device_spec.string_merge(op)
4541 if device_string is not prior_device_string:
4542 op._set_device_from_string(device_string)
4543 prior_device_string = device_string
|
| /external/ltp/testcases/kernel/fs/scsi/ltpscsi/ |
| D | scsimain.c | 588 char device_string[25]; in validate_device() local
591 sprintf(device_string, "/dev/sg%d", i); in validate_device()
593 if (strcmp(device, device_string) == 0) { in validate_device()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1118 bool IsTpuDevice(const string& device_string) { in IsTpuDevice() argument
1120 return DeviceNameUtils::ParseFullName(device_string, &device) && in IsTpuDevice()
|