diff --git a/tensorflow/lite/delegates/flex/kernel.cc b/tensorflow/lite/delegates/flex/kernel.cc index ceb9918f6fa..6443e9fe1bf 100644 --- a/tensorflow/lite/delegates/flex/kernel.cc +++ b/tensorflow/lite/delegates/flex/kernel.cc @@ -558,9 +558,16 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { } // namespace kernel TfLiteRegistration GetKernel() { - TfLiteRegistration registration{&kernel::Init, &kernel::Free, - &kernel::Prepare, &kernel::Eval, - nullptr, kTfLiteBuiltinDelegate}; + TfLiteRegistration registration{ + &kernel::Init, + &kernel::Free, + &kernel::Prepare, + &kernel::Eval, + nullptr, // .profiling_string + kTfLiteBuiltinDelegate, // .builtin_code + "TfLiteFlexDelegate", // .custom_name + 1, // .version + }; return registration; } diff --git a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc index 0a5a45561db..d53882797bb 100644 --- a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc +++ b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc @@ -1391,6 +1391,8 @@ TfLiteDelegate* NnApiDelegate() { .profiling_string = nullptr, .builtin_code = kTfLiteBuiltinDelegate, + .custom_name = "TfLiteNnapiDelegate", + .version = 1, }; // Request TFLite to partition the graph and make kernels