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