Searched refs:host_cpu_device_ (Results 1 – 4 of 4) sorted by relevance
121 input_devices_[i] = host_cpu_device_; in Init()142 function_def = pflr_->GetFLR(host_cpu_device_->name()) in InstantiateFunc()437 return host_cpu_device_; in InputDevice()
118 host_cpu_device_(host_cpu_device), in KernelAndDevice()181 Device* const host_cpu_device_; // non-null variable
200 Device* HostCPU() const { return host_cpu_device_; } in HostCPU()588 Device* host_cpu_device_; // Owned by device_manager variable
85 host_cpu_device_(device_mgr->HostCPU()), in EagerContext()1077 host_cpu_device_ = local_device_manager_.Get()->HostCPU(); in StoreCollectiveOpsServer()1322 host_cpu_device_ = local_device_manager_.Get()->HostCPU(); in SetMasterContextState()