Register reduce_xx operator for internal tflite interpreter (#3785)
author이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Mon, 3 Dec 2018 01:31:13 +0000 (10:31 +0900)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Mon, 3 Dec 2018 01:31:12 +0000 (10:31 +0900)
Reduce_sum, reduce_min, reduce_max builtin operators are registered
for internal tflite interpreter.

Signed-off-by: Chunseok Lee <chunseok.lee@samsung.com>
libs/support/tflite/src/kernels/register.cpp

index eb09c4e..cb48ff6 100644 (file)
@@ -92,6 +92,9 @@ TfLiteRegistration *Register_SIN();
 TfLiteRegistration *Register_TRANSPOSE_CONV();
 TfLiteRegistration *Register_SPARSE_TO_DENSE();
 #ifndef OBS_BUILD
+TfLiteRegistration *Register_SUM();
+TfLiteRegistration *Register_REDUCE_MAX();
+TfLiteRegistration *Register_REDUCE_MIN();
 TfLiteRegistration *Register_EQUAL();
 TfLiteRegistration *Register_NOT_EQUAL();
 TfLiteRegistration *Register_SQRT();
@@ -175,6 +178,9 @@ BuiltinOpResolver::BuiltinOpResolver()
   AddBuiltin(BuiltinOperator_SLICE, Register_SLICE());
   AddBuiltin(BuiltinOperator_SIN, Register_SIN());
 #ifndef OBS_BUILD
+  AddBuiltin(BuiltinOperator_SUM, Register_SUM());
+  AddBuiltin(BuiltinOperator_REDUCE_MAX, Register_REDUCE_MAX());
+  AddBuiltin(BuiltinOperator_REDUCE_MIN, Register_REDUCE_MIN());
   AddBuiltin(BuiltinOperator_TRANSPOSE_CONV, Register_TRANSPOSE_CONV());
   AddBuiltin(BuiltinOperator_SPARSE_TO_DENSE, Register_SPARSE_TO_DENSE());
   AddBuiltin(BuiltinOperator_EQUAL, Register_EQUAL());