Searched refs:cpu_dev (Results 1 – 3 of 3) sorted by relevance
122 Device* cpu_dev = nullptr; in RecvFromPeer() local123 Status status = dev_mgr_->LookupDevice("CPU:0", &cpu_dev); in RecvFromPeer()131 Tensor* cpu_tensor = new Tensor(cpu_dev->GetAllocator(cpu_attr), in RecvFromPeer()136 nullptr /*send_dev_ctx*/, to_device_ctx, cpu_dev, in RecvFromPeer()
587 Device* cpu_dev = nullptr; in RecvBufAsync() local588 s = env_->device_mgr->LookupDevice("CPU:0", &cpu_dev); in RecvBufAsync()593 Tensor* cpu_tensor = new Tensor(cpu_dev->GetAllocator(cpu_attr), in RecvBufAsync()
1298 cpu_dev = device_util.canonicalize("CPU:0")1300 devices = [cpu_dev, gpu_dev]