Searched defs:host_cpu_device (Results 1 – 3 of 3) sorted by relevance
122 Device* host_cpu_device) in KernelAndDevice()205 Device* host_cpu_device) in KernelAndDeviceOp()278 Device* host_cpu_device, const string& name, in KernelAndDeviceFunc()
35 Device* host_cpu_device) in TestKernelAndDeviceFunc()
3007 const string& host_cpu_device, in CreateTpuExecuteDummyArg()3152 const string& host_cpu_device, int64_t var_index, in CreateOrGetPerHostVariableCopy()