MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / OnSameTask

Method OnSameTask

tensorflow/core/common_runtime/eager/context.cc:517–523  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

515}
516
517bool EagerContext::OnSameTask(const Device* first, const Device* second) const {
518 if (first == nullptr) first = HostCPU();
519 if (second == nullptr) second = HostCPU();
520 return first->parsed_name().job == second->parsed_name().job &&
521 first->parsed_name().replica == second->parsed_name().replica &&
522 first->parsed_name().task == second->parsed_name().task;
523}
524
525// Gets the CPU device on the task of device.
526Status EagerContext::CPUDeviceOnTask(const Device* device,

Callers 1

EagerRemoteExecuteFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected