Home
last modified time | relevance | path

Searched refs:SetSequencingEvent (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer.h66 void SetSequencingEvent(EventPool::Handle event, se::Stream* stream);
Dtracked_device_buffer.cc32 void BufferSequencingEvent::SetSequencingEvent(EventPool::Handle event, in SetSequencingEvent() function in xla::BufferSequencingEvent
Dpjrt_stream_executor_client.cc405 definition_events.back()->SetSequencingEvent( in AllocateDestinationBuffer()
449 definition_events.back()->SetSequencingEvent(event_or.ConsumeValueOrDie(), in AllocateDestinationBuffer()
490 definition_event->SetSequencingEvent(event_or.ConsumeValueOrDie(), in AddDestinationBufferSynchronization()
1252 usage_event->SetSequencingEvent(event_or.ConsumeValueOrDie(), stream); in ToLiteral()
1544 transfer_event->SetSequencingEvent(event_or.ConsumeValueOrDie(), stream); in MakeTupleHelper()
1918 definition_event->SetSequencingEvent(event_or.ConsumeValueOrDie(), stream); in ExecuteHelper()