Searched refs:GetDeviceContextLocked (Results 1 – 2 of 2) sorted by relevance
260 return GetDeviceContextLocked().status(); in EnsureDeviceContextOk()278 xla::StatusOr<XlaDeviceContext*> XlaDevice::GetDeviceContextLocked() { in GetDeviceContextLocked() function in tensorflow::XlaDevice354 return GetDeviceContextLocked().status(); in UseGpuDeviceInfo()362 GetDeviceContextLocked()); in FillContextMap()457 GetDeviceContextLocked()); in MakeTensorFromProto()
187 xla::StatusOr<XlaDeviceContext*> GetDeviceContextLocked()