Searched refs:CPUDeviceOnTask (Results 1 – 3 of 3) sorted by relevance
421 Status CPUDeviceOnTask(const Device* device, Device** cpu_device) const;
674 ctx.CPUDeviceOnTask(op->Device(), &remote_cpu_device)); in EagerRemoteExecute()
704 Status EagerContext::CPUDeviceOnTask(const Device* device, in CPUDeviceOnTask() function in tensorflow::EagerContext