Home
last modified time | relevance | path

Searched refs:EnsureStreamOkLocked (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc277 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()
Dxla_device.h195 Status EnsureStreamOkLocked(xla::Backend* backend, const string& name,