| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils.cc | 230 *canonical_name = ParsedNameToString(parsed_name); in CanonicalizeDeviceName()
235 *canonical_name = ParsedNameToString(parsed_name); in CanonicalizeDeviceName()
244 string DeviceNameUtils::ParsedNameToString(const ParsedName& pn) { in ParsedNameToString() function in tensorflow::DeviceNameUtils
350 const auto& ParsedNameToString = DeviceNameUtils::ParsedNameToString; in MergeDevNamesImpl() local
355 ParsedNameToString(*target), "' and '", ParsedNameToString(other), in MergeDevNamesImpl()
367 ParsedNameToString(*target), "' and '", ParsedNameToString(other), in MergeDevNamesImpl()
379 ParsedNameToString(*target), "' and '", ParsedNameToString(other), in MergeDevNamesImpl()
392 ParsedNameToString(*target), "' and '", ParsedNameToString(other), in MergeDevNamesImpl()
412 ParsedNameToString(*target), "' and '", ParsedNameToString(other), in MergeDevNamesImpl()
620 *host_device_name = DeviceNameUtils::ParsedNameToString(device); in DeviceNameToCpuDeviceName()
[all …]
|
| D | device_name_utils_test.cc | 33 string round_tripped = DeviceNameUtils::ParsedNameToString(p); in RoundTripParsedName()
440 << "parsed_target: " << DeviceNameUtils::ParsedNameToString(parsed_target) in MergeOverrideHelper()
442 << DeviceNameUtils::ParsedNameToString(parsed_expected); in MergeOverrideHelper()
|
| D | device_name_utils.h | 201 static std::string ParsedNameToString(const ParsedName& pn);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 65 cpu_device_ = DeviceNameUtils::ParsedNameToString(parsed_name); in RecvAtHostOp()
69 tpu_device_ = DeviceNameUtils::ParsedNameToString(parsed_name); in RecvAtHostOp()
95 tpu_device = DeviceNameUtils::ParsedNameToString(parsed_name); in ComputeAsync()
222 cpu_device_ = DeviceNameUtils::ParsedNameToString(parsed_name); in SendFromHostOp()
226 tpu_device_ = DeviceNameUtils::ParsedNameToString(parsed_name); in SendFromHostOp()
250 tpu_device = DeviceNameUtils::ParsedNameToString(parsed_name); in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_structs.cc | 28 gpu_metadata_.insert({DeviceNameUtils::ParsedNameToString(device), metadata}); in AddGpuDevice()
33 auto it = gpu_metadata_.find(DeviceNameUtils::ParsedNameToString(device)); in GetGpuDeviceMetadata()
|
| D | tf_ops.cc | 125 context, DeviceNameUtils::ParsedNameToString(ndef_parsed_device)); in MultiDeviceProcessInlinedCallBlocks()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 277 DeviceNameUtils::ParsedNameToString(src_root.assigned_device_name_), in EnsureCompatibilityAcrossResourceEdge()
278 " vs ", DeviceNameUtils::ParsedNameToString(assigned_device_name_), in EnsureCompatibilityAcrossResourceEdge()
289 DeviceNameUtils::ParsedNameToString(src_root.resource_device_name_), in EnsureCompatibilityAcrossResourceEdge()
290 " vs ", DeviceNameUtils::ParsedNameToString(resource_device_name_), in EnsureCompatibilityAcrossResourceEdge()
307 << DeviceNameUtils::ParsedNameToString(requested_device_name_) in EnsureCompatibilityAcrossResourceEdge()
312 << DeviceNameUtils::ParsedNameToString( in EnsureCompatibilityAcrossResourceEdge()
502 DeviceNameUtils::ParsedNameToString(assigned_device_name_), in AssignDevice()
511 DeviceNameUtils::ParsedNameToString(resource_device_name_), in AssignDevice()
520 DeviceNameUtils::ParsedNameToString(requested_device_name_), in AssignDevice()
571 DeviceNameUtils::ParsedNameToString(requested_device_name_), in DebugString()
[all …]
|
| D | inspecting_placer.cc | 54 DeviceNameUtils::ParsedNameToString(devices.requested_device_name), in DebugString()
56 DeviceNameUtils::ParsedNameToString(devices.resource_device_name), in DebugString()
|
| D | composite_device.cc | 31 const string device_name = DeviceNameUtils::ParsedNameToString(parsed_name); in MakeDevice()
|
| D | collective_param_resolver_local.cc | 402 DeviceNameUtils::ParsedNameToString(parsed_name); in OrderTaskDeviceMap()
420 next_device = DeviceNameUtils::ParsedNameToString(parsed_name); in OrderTaskDeviceMap()
|
| D | inline_function_utils.cc | 238 return DeviceNameUtils::ParsedNameToString(ndef_parsed_device); in BodyNodeDevice()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 168 return DeviceNameUtils::ParsedNameToString(system_device); in GetTPUCompilationDevice()
175 return DeviceNameUtils::ParsedNameToString(tpu_device); in GetCPUHostDeviceForTPUDevice()
206 /*device=*/tensorflow::DeviceNameUtils::ParsedNameToString(tpu_device), in GetFullMeshTPUExecutionDeviceAssignment()
410 device_and_host.device = DeviceNameUtils::ParsedNameToString(tpu_device); in GetGeneralTPUExecutionDeviceAssignment()
|
| D | device_util.cc | 128 string name = DeviceNameUtils::ParsedNameToString(device->parsed_name()); in AddDevicesToOp()
|
| D | device_util_test.cc | 174 ASSERT_EQ(DeviceNameUtils::ParsedNameToString(devices.device_names()[0]), in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.cc | 70 DeviceNameUtils::ParsedNameToString(context_->HostCPUParsedName()) in Resolve()
|
| D | eager_operation.cc | 459 device_name_ = DeviceNameUtils::ParsedNameToString(device_parsed_name_); in SetDeviceName()
|
| D | context.cc | 1256 DeviceNameUtils::ParsedNameToString(device_name)); in GetClient()
1271 DeviceNameUtils::ParsedNameToString(device_name)); in GetClient()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | cross_device_transfer.cc | 72 return DeviceNameUtils::ParsedNameToString(parsed_name); in CanonicalizeDeviceName()
|
| D | remote_run_encapsulate.cc | 172 &getContext(), DeviceNameUtils::ParsedNameToString(parsed_name)); in runOnOperation()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 186 devices_.insert(DeviceNameUtils::ParsedNameToString(name)); in AddDevice()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 267 DeviceNameUtils::ParsedNameToString(segment_device)); in GetEngineInfo()
269 info->device = DeviceNameUtils::ParsedNameToString(segment_device); in GetEngineInfo()
280 info->device = DeviceNameUtils::ParsedNameToString(parsed_name); in GetEngineInfo()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.cc | 118 << DeviceNameUtils::ParsedNameToString(host_spec); in GetHostSystemDevices()
|
| D | configure_tpu_embedding_rewrite_pass.cc | 234 return DeviceNameUtils::ParsedNameToString(device_spec); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_tf_ops_pass.cc | 61 std::string result = DeviceNameUtils::ParsedNameToString( in GetHost()
|
| D | tfg-to-tfe.cc | 99 tensorflow::DeviceNameUtils::ParsedNameToString(parsed_name)))); in ReformatOpAttributes()
|