Searched refs:HostCPU (Results 1 – 5 of 5) sorted by relevance
71 device = ctx->HostCPU(); in StepStatsDeviceIndex()99 handle_device == nullptr ? ctx->HostCPU() : handle_device; in MaybeCopyInputToExpectedDevice()252 if (first == nullptr) first = ctx->HostCPU(); in OnSameTask()253 if (second == nullptr) second = ctx->HostCPU(); in OnSameTask()277 *device = ctx->HostCPU(); in FindDeviceFromName()325 Device* cpu_device = ctx->HostCPU(); in AddInputDevicesToCacheKey()446 ctx->GetCollectiveExecutorHandle(), ctx->HostCPU()); in EagerLocalExecute()453 ctx->GetCollectiveExecutorHandle(), ctx->HostCPU()); in EagerLocalExecute()602 TF_RETURN_IF_ERROR(h->CopyToDevice(ctx, ctx->HostCPU(), &actual_handle)); in EagerRemoteSendTensor()823 Device* op_device = op->Device() == nullptr ? ctx->HostCPU() : op->Device(); in MaybeUpdateOpDevice()[all …]
125 (ctx_ == nullptr || ctx_->HostCPU() == device_); in OnHostCPU()
143 Device* HostCPU() const { return devices_[0]; } in HostCPU() function
259 if (srcd == nullptr) srcd = ctx->HostCPU(); in CopyToDevice()
529 h->handle->Context()->HostCPU()->name().c_str(), &h_cpu); in TFE_TensorHandleResolve()544 h->handle->Context(), h->handle->Context()->HostCPU(), &h_cpu); in TFE_TensorHandleResolve()601 ? op->operation.EagerContext()->HostCPU() in TFE_OpGetDevice()