Searched refs:copy_stream (Results 1 – 3 of 3) sorted by relevance
72 Stream_t *copy_stream(Stream_t *Stream);82 copy_stream( (stream) )
36 Stream_t *copy_stream(Stream_t *Stream) in copy_stream() function
365 LocalDeviceState* local_device, se::Stream* copy_stream, in AllocateDestinationBuffer() argument381 if (copy_stream == nullptr) { in AllocateDestinationBuffer()384 copy_stream->ThenWaitFor(local_device->compute_stream()); in AllocateDestinationBuffer()425 if (tuple_table_stream != copy_stream) { in AllocateDestinationBuffer()483 se::Stream* copy_stream) { in AddDestinationBufferSynchronization() argument485 local_device->event_pool().ThenAllocateAndRecordEvent(copy_stream); in AddDestinationBufferSynchronization()487 StallStreamOnError(local_device, copy_stream); in AddDestinationBufferSynchronization()491 copy_stream); in AddDestinationBufferSynchronization()499 definition_event, copy_stream, in AddDestinationBufferSynchronization()