Searched refs:async_kernel (Results 1 – 3 of 3) sorted by relevance
623 AsyncOpKernel* async_kernel = item.kernel->AsAsync(); in ProcessAsync() local624 DCHECK(async_kernel != nullptr); in ProcessAsync()663 [async_kernel, state] { in ProcessAsync()664 return async_kernel->TraceString( in ProcessAsync()668 immutable_state_.params().device->ComputeAsync(async_kernel, &state->ctx, in ProcessAsync()
557 TfLiteAsyncKernel* (*async_kernel)(TfLiteOpaqueContext* context,
977 struct TfLiteAsyncKernel* (*async_kernel)(TfLiteContext* context, member