[Demo] Enable extended op for android jni (#1653)

This will enable our extended operators for TFLite JNI

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/externals/CMakeLists.txt b/externals/CMakeLists.txt
index 3527a77..4b71b9c 100644 (file)
--- a/externals/CMakeLists.txt
+++ b/externals/CMakeLists.txt
@@ -81,12 +81,9 @@ if("${TARGET_OS}" STREQUAL "android")
                       )
   set(TFLITE_JNI_INCLUDES ${TENSORFLOW_LITE_BASE}/java/src/native)
 
-  # We need this for running vanilla tflite
-  # TODO remove this when nnfw is used
-  set(TFLITE_SRCS_V ${TENSORFLOW_LITE_BASE}/kernels/register.cc)
-
   # TODO use tensorflow-lite static library instead of compiling all the sources again
   add_library(tensorflowlite_jni SHARED ${TFLITE_JNI_SRCS} ${TFLITE_SRCS} ${TFLITE_SRCS_V})
   target_include_directories(tensorflowlite_jni PUBLIC ${TFLITE_JNI_INCLUDES} ${TFLITE_INCLUDES})
   target_link_libraries(tensorflowlite_jni eigen3 ${LIB_PTHREAD} dl)
+  target_link_libraries(tensorflowlite_jni nnfw_support_tflite)
 endif()