Searched refs:CPUDeviceOnTask (Results 1 – 3 of 3) sorted by relevance
494 Status CPUDeviceOnTask(const Device* device, Device** cpu_device) const;
868 ctx.CPUDeviceOnTask(op_device, &remote_cpu_device)); in EagerRemoteExecute()
962 Status EagerContext::CPUDeviceOnTask(const Device* device, in CPUDeviceOnTask() function in tensorflow::EagerContext