Home
last modified time | relevance | path

Searched refs:block_host_until_done (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc130 const BufferAllocations& buffer_allocations, bool block_host_until_done, in ExecuteThunks() argument
226 if (do_profile || block_host_until_done) { in ExecuteThunks()
423 const bool block_host_until_done = in ExecuteAsyncOnStreamImpl() local
553 block_host_until_done, in ExecuteAsyncOnStreamImpl()
Dgpu_executable.h142 bool block_host_until_done,
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.h389 void (*block_host_until_done)(const SP_Device* device, SP_Stream stream, member
Dstream_executor.cc556 if (stream_executor_->block_host_until_done != nullptr) { in BlockHostUntilDone()
557 stream_executor_->block_host_until_done(&device_, stream_handle, in BlockHostUntilDone()
Dstream_executor_test.cc860 se_.block_host_until_done = [](const SP_Device* const device, in TEST_F()