Searched refs:GetTaskName (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/util/ |
D | device_name_utils.h | 194 static bool GetTaskName(const ParsedName& pn, std::string* task);
|
D | device_name_utils.cc | 546 bool DeviceNameUtils::GetTaskName(const ParsedName& pn, string* task) { in GetTaskName() function in tensorflow::DeviceNameUtils
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | session_mgr.cc | 237 DeviceNameUtils::GetTaskName(device->parsed_name(), &task_name); in UpdateSession()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | execute.cc | 591 if (!DeviceNameUtils::GetTaskName(output_device->parsed_name(), in CreateUnshapedOutput() 836 if (!DeviceNameUtils::GetTaskName(op->GetDeviceParsedName(), &remote_task)) { in EagerRemoteExecute() 1279 if (!DeviceNameUtils::GetTaskName(device->parsed_name(), &remote_task)) { in EagerCopyToDevice()
|
D | context.cc | 981 Status GetTaskName(Device* d, string* task_name) { in GetTaskName() function 1000 if (!DeviceNameUtils::GetTaskName(device_name, &device_task_name)) { in GetClient() 1133 if (!DeviceNameUtils::GetTaskName(pw, &remote_worker_task_name)) { in SetRemoteDeviceFilters()
|
D | context_distributed_manager.cc | 316 tensorflow::DeviceNameUtils::GetTaskName(pn, &task_name); in UpdateRemoteContexts()
|
D | tensor_handle.cc | 877 if (!DeviceNameUtils::GetTaskName(device->parsed_name(), &remote_task)) { in SetRemoteShapeAndDevice()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | collective_param_resolver_local.cc | 82 CHECK(DeviceNameUtils::GetTaskName(parsed_device, &task_name)); in TaskNameFromDeviceName()
|