Searched refs:GetDeviceContextLocked (Results 1 – 2 of 2) sorted by relevance
274 return GetDeviceContextLocked().status(); in EnsureDeviceContextOk()293 XlaDevice::GetDeviceContextLocked() { in GetDeviceContextLocked() function in tensorflow::XlaDevice380 return GetDeviceContextLocked().status(); in UseGpuDeviceInfo()386 TF_ASSIGN_OR_RETURN(auto device_contexts, GetDeviceContextLocked()); in TryGetDeviceContext()506 TF_ASSIGN_OR_RETURN(device_contexts, GetDeviceContextLocked()); in MakeTensorFromProto()519 TF_ASSIGN_OR_RETURN(device_contexts, GetDeviceContextLocked()); in MakeFastMemTensorFromProto()
202 GetDeviceContextLocked() TF_EXCLUSIVE_LOCKS_REQUIRED(mu_);