Searched refs:ThenRecordEvent (Results 1 – 15 of 15) sorted by relevance
54 void EventPool::ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle) { in ThenRecordEvent() function in xla::EventPool56 stream->ThenRecordEvent(handle.event_.get()); in ThenRecordEvent()64 ThenRecordEvent(stream, handle); in ThenAllocateAndRecordEvent()
84 void ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle);
1346 local_device->event_pool().ThenRecordEvent(stream, event_or.ValueOrDie()); in ToLiteral()
180 stream->ThenRecordEvent(e); in QueueInUse()
179 stream->ThenRecordEvent(e); in QueueInUse()
241 stream->ThenRecordEvent(definition_event.get()); in DoTpuExecute()
757 stream->ThenRecordEvent(definition_event.get()); in DoWork()
177 host_to_device_stream_->ThenRecordEvent(event.get()); in CopyCPUTensorToDevice()
274 dst_device_to_device_stream->ThenRecordEvent(definition_event.get()); in TpuDeviceToDeviceCopy()
494 stream->ThenRecordEvent(definition_event.get()); in PopulateOutputs()
300 async_comms_stream.ThenRecordEvent(&done_event); in RunNcclCollective()
1705 async_collectives->async_comm_stream()->ThenRecordEvent(&done_event); in operator ()()
417 stream.ThenRecordEvent(&event); in TEST_F()
245 Stream &ThenRecordEvent(Event *event);
315 Stream &Stream::ThenRecordEvent(Event *event) { in ThenRecordEvent() function in stream_executor::Stream