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