| /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()
591 *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 | 196 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 | 30 gpu_metadata_.insert({DeviceNameUtils::ParsedNameToString(device), metadata}); in AddGpuDevice()
35 auto it = gpu_metadata_.find(DeviceNameUtils::ParsedNameToString(device)); in GetGpuDeviceMetadata()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 256 DeviceNameUtils::ParsedNameToString(src_root.assigned_device_name_), in EnsureCompatibilityAcrossResourceEdge()
257 " vs ", DeviceNameUtils::ParsedNameToString(assigned_device_name_), in EnsureCompatibilityAcrossResourceEdge()
267 DeviceNameUtils::ParsedNameToString(src_root.resource_device_name_), in EnsureCompatibilityAcrossResourceEdge()
268 " vs ", DeviceNameUtils::ParsedNameToString(resource_device_name_), in EnsureCompatibilityAcrossResourceEdge()
284 << DeviceNameUtils::ParsedNameToString(requested_device_name_) in EnsureCompatibilityAcrossResourceEdge()
289 << DeviceNameUtils::ParsedNameToString( in EnsureCompatibilityAcrossResourceEdge()
479 DeviceNameUtils::ParsedNameToString(assigned_device_name_), in AssignDevice()
488 DeviceNameUtils::ParsedNameToString(resource_device_name_), in AssignDevice()
497 DeviceNameUtils::ParsedNameToString(requested_device_name_), in AssignDevice()
548 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 | 382 DeviceNameUtils::ParsedNameToString(parsed_name); in OrderTaskDeviceMap()
400 next_device = DeviceNameUtils::ParsedNameToString(parsed_name); in OrderTaskDeviceMap()
|
| D | inline_function_utils.cc | 250 return DeviceNameUtils::ParsedNameToString(ndef_parsed_device); in BodyNodeDevice()
|
| D | graph_execution_state.cc | 371 DeviceNameUtils::ParsedNameToString(parsed_name)); in LookupDevice()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 172 return DeviceNameUtils::ParsedNameToString(system_device); in GetTPUCompilationDevice()
179 return DeviceNameUtils::ParsedNameToString(tpu_device); in GetCPUHostDeviceForTPUDevice()
210 /*device=*/tensorflow::DeviceNameUtils::ParsedNameToString(tpu_device), in GetFullMeshTPUExecutionDeviceAssignment()
414 device_and_host.device = DeviceNameUtils::ParsedNameToString(tpu_device); in GetGeneralTPUExecutionDeviceAssignment()
|
| D | device_util.cc | 131 string name = DeviceNameUtils::ParsedNameToString(device->parsed_name()); in AddDevicesToOp()
|
| D | device_util_test.cc | 172 ASSERT_EQ(DeviceNameUtils::ParsedNameToString(devices.device_names()[0]), in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.cc | 62 DeviceNameUtils::ParsedNameToString(context_->HostCPUParsedName()) in Resolve()
|
| D | eager_operation.cc | 457 device_name_ = DeviceNameUtils::ParsedNameToString(device_parsed_name_); in SetDeviceName()
|
| D | context.cc | 1003 DeviceNameUtils::ParsedNameToString(device_name)); in GetClient()
1018 DeviceNameUtils::ParsedNameToString(device_name)); in GetClient()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_compile_op_replication_pass.cc | 39 return DeviceNameUtils::ParsedNameToString( in GetHost()
|
| D | cross_host_transfer.cc | 52 auto host = DeviceNameUtils::ParsedNameToString(parsed_name); in GetHost()
|
| D | cluster_tf_ops_pass.cc | 60 std::string result = DeviceNameUtils::ParsedNameToString( in GetHost()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 265 DeviceNameUtils::ParsedNameToString(segment_device)); in GetEngineInfo()
267 info->device = DeviceNameUtils::ParsedNameToString(segment_device); in GetEngineInfo()
278 info->device = DeviceNameUtils::ParsedNameToString(parsed_name); in GetEngineInfo()
|
| D | trt_optimization_pass.cc | 140 << DeviceNameUtils::ParsedNameToString(dev->parsed_name()); in PrintDebugInfo()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 186 devices_.insert(DeviceNameUtils::ParsedNameToString(name)); in AddDevice()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.cc | 118 << DeviceNameUtils::ParsedNameToString(host_spec); in GetHostSystemDevices()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental.cc | 547 auto str = tensorflow::DeviceNameUtils::ParsedNameToString(address_space); in TFE_HostAddressSpace()
|