Searched refs:dest_stream (Results 1 – 2 of 2) sorted by relevance
139 se::DeviceMemoryBase dest, se::Stream* dest_stream) { in EnqueueCopy() argument145 if (src_stream == dest_stream) { in EnqueueCopy()146 dest_stream->ThenMemcpyD2D(&dest, src, src.size()); in EnqueueCopy()154 dest_stream->ThenWaitFor(src_stream).ThenMemcpyD2D(&dest, src, src.size()); in EnqueueCopy()155 src_stream->ThenWaitFor(dest_stream); in EnqueueCopy()
332 StreamPtr dest_stream = MemoryStream::CreateRef(&data, nullptr); in ExtractData() local333 if (src_stream && dest_stream) { in ExtractData()338 CHECK(dest_stream->WriteAllBlocking(buffer, read, nullptr)); in ExtractData()