Searched refs:device_context_ (Results 1 – 4 of 4) sorted by relevance
223 if (device_context_) { in ~XlaDevice()224 device_context_->Unref(); in ~XlaDevice()282 bool need_new_device_context = !device_context_; in GetDeviceContextLocked()311 return device_context_; in GetDeviceContextLocked()317 if (device_context_) { in GetDeviceContextLocked()318 device_context_->Unref(); in GetDeviceContextLocked()324 device_context_ = new XlaDeviceContext( in GetDeviceContextLocked()329 << device_context_; in GetDeviceContextLocked()341 gpu_device_info->default_context = device_context_; in GetDeviceContextLocked()348 return device_context_; in GetDeviceContextLocked()
231 XlaDeviceContext* device_context_ GUARDED_BY(mu_) = nullptr;
70 device_context_->CopyCPUTensorToDevice( in MakeTensorFromProto()83 device_context_->Ref(); in FillContextMap()84 (*device_context_map)[n->id()] = device_context_; in FillContextMap()
207 device_context_(ctx) { in SYCLDevice()227 SYCLDeviceContext* device_context_; // not owned variable