Searched refs:TpuDriver_EventAwait (Results 1 – 3 of 3) sorted by relevance
255 TPUDRIVER_CAPI_EXPORT extern PrototypeTpuDriver_EventAwait TpuDriver_EventAwait;296 PrototypeTpuDriver_EventAwait* TpuDriver_EventAwait; // NOLINT member
134 TpuStatus* status = driver_fn.TpuDriver_EventAwait(transfer_sum_event, in main()
70 auto tpu_status = driver_fn_->TpuDriver_EventAwait(event_, -1); in Await()78 auto tpu_status_or = driver_fn_->TpuDriver_EventAwait( in AwaitWithTimeout()