Searched refs:BlockUntilDone (Results 1 – 7 of 7) sorted by relevance
56 AsExecutorStream(stream)->BlockUntilDone(); in Memcpy()65 AsExecutorStream(stream)->BlockUntilDone(); in Memcpy()96 AsExecutorStream(dependent)->BlockUntilDone(); in CreateStreamDependency()111 AsExecutorStream(stream)->BlockUntilDone(); in BlockHostUntilDone()
40 void BlockUntilDone();
56 void HostStream::BlockUntilDone() { in BlockUntilDone() function in stream_executor::host::HostStream
168 AsHostStream(dependent)->BlockUntilDone(); in CreateStreamDependency()183 AsHostStream(stream)->BlockUntilDone(); in BlockHostUntilDone()
46 Status BlockUntilDone() const;
40 Status AsyncExecution::BlockUntilDone() const { in BlockUntilDone() function in xla::AsyncExecution
929 TF_RETURN_IF_ERROR(execution->BlockUntilDone()); in WaitForExecution()