Searched refs:EnsureStreamOkLocked (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 263 Status XlaDevice::EnsureStreamOkLocked(xla::Backend* backend, in EnsureStreamOkLocked() function in tensorflow::XlaDevice
283 TF_RETURN_IF_ERROR(EnsureStreamOkLocked(backend, "stream", &stream_, in GetDeviceContextLocked()
290 TF_RETURN_IF_ERROR(EnsureStreamOkLocked(backend, "host_to_device_stream", in GetDeviceContextLocked()
295 EnsureStreamOkLocked(backend, "device_to_device_stream", &stream, in GetDeviceContextLocked()
|
| D | xla_device.h | 183 Status EnsureStreamOkLocked(xla::Backend* backend, const string& name,
|