Home
last modified time | relevance | path

Searched refs:gpu_delegate_kernel (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Ddelegate.cc516 auto gpu_delegate_kernel = in DelegatePrepare() local
518 const auto status = gpu_delegate_kernel->Prepare(context, params); in DelegatePrepare()
524 return gpu_delegate_kernel.release(); in DelegatePrepare()
538 auto* gpu_delegate_kernel = GetDelegateKernel(node); in DelegatePrepare() local
539 const auto status = gpu_delegate_kernel->GetRequiredTemporaries( in DelegatePrepare()
H A Dmetal_delegate.mm678 auto* gpu_delegate_kernel = GetMetalDelegate(node);
680 gpu_delegate_kernel->GetRequiredTemporaries(context, node, &node->temporaries);