Home
last modified time | relevance | path

Searched refs:GetTaskName (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils.h194 static bool GetTaskName(const ParsedName& pn, std::string* task);
Ddevice_name_utils.cc546 bool DeviceNameUtils::GetTaskName(const ParsedName& pn, string* task) { in GetTaskName() function in tensorflow::DeviceNameUtils
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc237 DeviceNameUtils::GetTaskName(device->parsed_name(), &task_name); in UpdateSession()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc591 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()
Dcontext.cc981 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()
Dcontext_distributed_manager.cc316 tensorflow::DeviceNameUtils::GetTaskName(pn, &task_name); in UpdateRemoteContexts()
Dtensor_handle.cc877 if (!DeviceNameUtils::GetTaskName(device->parsed_name(), &remote_task)) { in SetRemoteShapeAndDevice()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc82 CHECK(DeviceNameUtils::GetTaskName(parsed_device, &task_name)); in TaskNameFromDeviceName()