Searched refs:EventPool (Results 1 – 6 of 6) sorted by relevance
24 EventPool::Handle::~Handle() { in ~Handle()31 EventPool::EventPool(bool allow_reuse) in EventPool() function in xla::EventPool34 StatusOr<EventPool::Handle> EventPool::AllocateEvent( in AllocateEvent()53 void EventPool::ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle) { in ThenRecordEvent()59 StatusOr<EventPool::Handle> EventPool::ThenAllocateAndRecordEvent( in ThenAllocateAndRecordEvent()61 TF_ASSIGN_OR_RETURN(EventPool::Handle handle, in ThenAllocateAndRecordEvent()
29 class EventPool {56 friend class EventPool;58 EventPool* pool_ = nullptr;66 explicit EventPool(bool allow_reuse);83 void ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle);
104 EventPool& event_pool() { return event_pool_; } in event_pool()168 EventPool event_pool_;
66 void SetSequencingEvent(EventPool::Handle event, se::Stream* stream);107 EventPool::Handle event_;
32 void BufferSequencingEvent::SetSequencingEvent(EventPool::Handle event, in SetSequencingEvent()
402 TF_ASSIGN_OR_RETURN(EventPool::Handle event, in AllocateDestinationBuffer()442 StatusOr<EventPool::Handle> event_or = in AllocateDestinationBuffer()484 StatusOr<EventPool::Handle> event_or = in AddDestinationBufferSynchronization()1237 StatusOr<EventPool::Handle> event_or = in ToLiteral()1536 StatusOr<EventPool::Handle> event_or = in MakeTupleHelper()1903 StatusOr<EventPool::Handle> event_or = in ExecuteHelper()