Searched refs:kernel_registration (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate.cc | 32 TfLiteRegistration kernel_registration; in GetDelegateKernelRegistration() local
33 kernel_registration.profiling_string = nullptr; in GetDelegateKernelRegistration()
34 kernel_registration.builtin_code = kTfLiteBuiltinDelegate; in GetDelegateKernelRegistration()
35 kernel_registration.custom_name = delegate->Name(); in GetDelegateKernelRegistration()
36 kernel_registration.version = 1; in GetDelegateKernelRegistration()
37 kernel_registration.free = [](TfLiteContext* context, void* buffer) -> void { in GetDelegateKernelRegistration()
40 kernel_registration.init = [](TfLiteContext* context, const char* buffer, in GetDelegateKernelRegistration()
57 kernel_registration.prepare = [](TfLiteContext* context, in GetDelegateKernelRegistration()
67 kernel_registration.invoke = [](TfLiteContext* context, in GetDelegateKernelRegistration()
75 return kernel_registration; in GetDelegateKernelRegistration()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate.mm | 197 TfLiteRegistration kernel_registration;
198 kernel_registration.builtin_code = kTfLiteBuiltinDelegate;
199 kernel_registration.custom_name = "TfLiteCoreMlDelegate";
200 kernel_registration.free = [](TfLiteContext* context, void* buffer) -> void {
203 kernel_registration.init = [](TfLiteContext* context, const char* buffer,
214 kernel_registration.invoke = [](TfLiteContext* context, TfLiteNode* node) -> TfLiteStatus {
222 kernel_registration.prepare = [](TfLiteContext* context, TfLiteNode* node) -> TfLiteStatus {
231 return kernel_registration;
|