Searched refs:delegate_kernel (Results 1 – 3 of 3) sorted by relevance
51 std::unique_ptr<SimpleDelegateKernelInterface> delegate_kernel( in GetDelegateKernelRegistration() local53 if (delegate_kernel->Init(context, params) != kTfLiteOk) { in GetDelegateKernelRegistration()56 return delegate_kernel.release(); in GetDelegateKernelRegistration()64 SimpleDelegateKernelInterface* delegate_kernel = in GetDelegateKernelRegistration() local66 return delegate_kernel->Prepare(context, node); in GetDelegateKernelRegistration()70 SimpleDelegateKernelInterface* delegate_kernel = in GetDelegateKernelRegistration() local72 TFLITE_DCHECK(delegate_kernel != nullptr); in GetDelegateKernelRegistration()73 return delegate_kernel->Eval(context, node); in GetDelegateKernelRegistration()
71 def DelegateKernelOp : RuntimeFallbackDialect_Op<"delegate_kernel"> {74 The "tfd.delegate_kernel" operation takes an input chain, and arbitrary95 %out_c, %out_tensor = "tfd.delegate_kernel"(
143 DelegateKernel* delegate_kernel = in TEST_F() local145 const auto& tensor_release_map = GetTensorReleaseMap(delegate_kernel); in TEST_F()