Searched refs:device_string (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_collective_permute_thunk.cc | 131 std::string device_string = GetDeviceString(params.nccl_params); in RunNcclCollective() local
138 comm, device_string, current_id); in RunNcclCollective()
144 absl::string_view device_string, int64_t current_id) { in RunCollectivePermute() argument
196 device_string, current_id, in RunCollectivePermute()
213 device_string, src_addr.opaque(), element_count, *target_id, in RunCollectivePermute()
224 device_string, dest_addr.opaque(), element_count, *source_id, in RunCollectivePermute()
235 device_string); in RunCollectivePermute()
|
| D | nccl_collective_permute_thunk.h | 98 absl::string_view device_string, int64_t current_id);
|
| D | jitrt_custom_calls.cc | 2037 std::string device_string = NcclCollectiveThunk::GetDeviceString(params); in operator ()() local
2052 device_string, current_id); in operator ()()
|
| /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/core/common_runtime/ |
| D | lower_functional_ops.cc | 91 bool IsPropagatableDevice(StringPiece device_string) { in IsPropagatableDevice() argument
93 return DeviceNameUtils::ParseFullName(device_string, &device) && in IsPropagatableDevice()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 1066 def check_device(path, device_string): argument
1067 spec = tf_device.DeviceSpec.from_string(device_string)
1075 "enqueue method for an example.".format(path, device_string))
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 53 def set_cpu0(device_string): argument
64 parsed_device = pydev.DeviceSpec.from_string(device_string)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4796 device_string = device_spec.string_merge(op)
4800 if device_string is not prior_device_string:
4801 op._set_device_from_string(device_string)
4802 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 | 1368 bool IsTpuDevice(StringPiece device_string) { in IsTpuDevice() argument
1370 return DeviceNameUtils::ParseFullName(device_string, &device) && in IsTpuDevice()
|