Enable DETECTION_POSTPROCESS kernel in TFLite interpreter (#6214)

- Enable DETECTION_POSTPROCESS in interpreter

Signed-off-by: Chunseok Lee <chunseok.lee@samsung.com>

diff --git a/runtimes/libs/tflite/src/ext/kernels/register.cpp b/runtimes/libs/tflite/src/ext/kernels/register.cpp
index cf2c81f..a99899a 100644 (file)
--- a/runtimes/libs/tflite/src/ext/kernels/register.cpp
+++ b/runtimes/libs/tflite/src/ext/kernels/register.cpp
@@ -24,6 +24,16 @@ limitations under the License.
 
 namespace tflite {
 namespace ops {
+namespace custom {
+
+TfLiteRegistration* Register_DETECTION_POSTPROCESS();
+
+}  // namespace custom
+}
+}
+
+namespace tflite {
+namespace ops {
 namespace builtin {
 
 TfLiteRegistration *Register_RELU();
@@ -128,6 +138,7 @@ namespace tflite {
 BuiltinOpResolver::BuiltinOpResolver()
 {
   // Using namespace directive to minimize diff with upstream tensorflow
+  using namespace ::tflite::ops::custom;
   using namespace ::tflite::ops::builtin;
   using namespace ::tflite;
 
@@ -227,6 +238,9 @@ BuiltinOpResolver::BuiltinOpResolver()
   AddCustom("SquaredDifference", nnfw::tflite::custom::Register_SquaredDifference());
   AddCustom("TensorFlowSum", nnfw::tflite::custom::Register_TensorFlowSum());
   AddCustom("Abs", nnfw::tflite::custom::Register_Abs());
+  AddCustom("TFLite_Detection_PostProcess",
+            ::tflite::ops::custom::Register_DETECTION_POSTPROCESS());
+
 }
 
 }  // namespace tflite