Searched refs:LookupEvent (Results 1 – 3 of 3) sorted by relevance
134 auto se_event = tpu_platform().LookupEvent(event->implementation()); in PollForEventStatus()143 auto se_event = tpu_platform().LookupEvent(event->implementation()); in RecordEvent()153 auto se_event = tpu_platform().LookupEvent(event->implementation()); in WaitForEvent()
123 SE_Event* LookupEvent(stream_executor::internal::EventInterface* key);
142 SE_Event* TpuPlatform::LookupEvent( in LookupEvent() function in tensorflow::tpu::TpuPlatform