Fix custom operator register macro (#3495)
author오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Wed, 7 Nov 2018 06:34:34 +0000 (15:34 +0900)
committer이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Wed, 7 Nov 2018 06:34:34 +0000 (15:34 +0900)
Fix custom operator register macro for updated TfLiteRegistration field
- First 4 fields may not change
- Can adjust new fields before custom_name

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
include/support/tflite/kernels/CustomOps.h

index aceb835..f0c4419 100644 (file)
@@ -44,7 +44,8 @@ namespace nnfw
   TfLiteRegistration *Register_##Name(void)                                                      \
   {                                                                                              \
     static TfLiteRegistration r = { Name::Init##Name , Name::Free##Name , Name::Prepare##Name ,  \
-                                   Name::Eval##Name , 0, #Name};                                 \
+                                   Name::Eval##Name ,};                                          \
+    r.custom_name = #Name;                                                                       \
     return &r;                                                                                   \
   }