Searched refs:device_context_ (Results 1 – 5 of 5) sorted by relevance
231 if (device_context_) { in ~XlaDevice()232 device_context_->Unref(); in ~XlaDevice()298 bool need_new_device_context = !device_context_; in GetDeviceContextLocked()327 return std::make_pair(device_context_, fast_mem_device_context_); in GetDeviceContextLocked()333 if (device_context_) { in GetDeviceContextLocked()334 device_context_->Unref(); in GetDeviceContextLocked()343 device_context_ = new XlaDeviceContext( in GetDeviceContextLocked()348 << device_context_; in GetDeviceContextLocked()367 gpu_device_info->default_context = device_context_; in GetDeviceContextLocked()374 return std::make_pair(device_context_, fast_mem_device_context_); in GetDeviceContextLocked()
250 XlaDeviceContext* device_context_ TF_GUARDED_BY(mu_) = nullptr;
339 DeviceContext* device_context_ = nullptr; member in tensorflow::__anonce8950920111::ExecutorState427 if (device_context_) { in ~ExecutorState()428 device_context_->Unref(); in ~ExecutorState()464 device->TryGetDeviceContext(&device_context_); in RunAsync()735 params.op_device_context = device_context_; in Process()
399 device_context_->Unref(); in ~BaseGPUDevice()436 device_context_ = in Init()474 gpu_device_info_->default_context = device_context_; in Init()545 GPUDeviceContext* gpu_device_context = device_context_; in Compute()618 GPUDeviceContext* gpu_device_context = device_context_; in ComputeAsync()679 device_context_->CopyCPUTensorToDevice( in MaybeCopyTensorToGPU()
146 GPUDeviceContext* device_context_; variable