Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/
Ddelegate.cc375 auto gpu_delegate_kernel = in DelegatePrepare() local
377 const auto status = gpu_delegate_kernel->Prepare(context, params); in DelegatePrepare()
383 return gpu_delegate_kernel.release(); in DelegatePrepare()
397 auto* gpu_delegate_kernel = GetDelegateKernel(node); in DelegatePrepare() local
398 const auto status = gpu_delegate_kernel->GetRequiredTemporaries( in DelegatePrepare()
Dmetal_delegate.mm656 auto* gpu_delegate_kernel = GetMetalDelegate(node);
658 gpu_delegate_kernel->GetRequiredTemporaries(context, node, &node->temporaries);