Home
last modified time | relevance | path

Searched refs:RefreshStatus (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dlocal_device_state.cc131 auto status = stream->RefreshStatus(); // Can return error::Unimplemented in BorrowStreamFromPool()
141 auto status = stream->RefreshStatus(); // Can return error::Unimplemented in ReturnStreamToPool()
Dpjrt_stream_executor_client.cc912 h2d_stream->RefreshStatus() in BufferFromHostLiteral()
/external/tensorflow/tensorflow/core/framework/
Ddevice.h126 virtual Status RefreshStatus() { in RefreshStatus() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tensor.cc101 status.Update(stream->RefreshStatus()); in RefreshStatusOfStreams()
Dxla_device.h189 Status RefreshStatus() override TF_LOCKS_EXCLUDED(mu_);
Dxla_device.cc552 Status XlaDevice::RefreshStatus() { in RefreshStatus() function in tensorflow::XlaDevice
561 Status status = stream->RefreshStatus(); in RefreshStatus()
Dxla_device_context.cc278 !device_to_host_stream->RefreshStatus().ok()) { in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc502 TF_ASSERT_OK(stream.RefreshStatus()); in TEST_F()
504 auto updated_status = stream.RefreshStatus(); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc1262 status.Update(device->RefreshStatus()); in Finish()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h135 port::Status RefreshStatus() TF_LOCKS_EXCLUDED(mu_);
Dstream.cc286 port::Status Stream::RefreshStatus() { in RefreshStatus() function in stream_executor::Stream