Ensure that delegate kernels have valid custom names
PiperOrigin-RevId: 239268129
This commit is contained in:
parent
e028703c1c
commit
efb872186d
@ -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;
|
||||
}
|
||||
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user