Searched refs:ThenRecordEvent (Results 1 – 13 of 13) sorted by relevance
53 void EventPool::ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle) { in ThenRecordEvent() function in xla::EventPool55 stream->ThenRecordEvent(handle.event_.get()); in ThenRecordEvent()63 ThenRecordEvent(stream, handle); in ThenAllocateAndRecordEvent()
83 void ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle);
1251 local_device->event_pool().ThenRecordEvent(stream, event_or.ValueOrDie()); in ToLiteral()
179 stream->ThenRecordEvent(e); in QueueInUse()
180 stream->ThenRecordEvent(e); in QueueInUse()
175 host_to_device_stream_->ThenRecordEvent(event.get()); in CopyCPUTensorToDevice()
266 dst_device_to_device_stream->ThenRecordEvent(definition_event.get()); in TpuDeviceToDeviceCopy()
458 stream->ThenRecordEvent(definition_event.get()); in PopulateOutputs()
203 stream->ThenRecordEvent(finish_event.get()); in ExecuteThunks()
762 stream->ThenRecordEvent(definition_event.get()); in DoWork()
593 stream.ThenRecordEvent(&event); in TEST_F()
239 Stream &ThenRecordEvent(Event *event);
329 Stream &Stream::ThenRecordEvent(Event *event) { in ThenRecordEvent() function in stream_executor::Stream