[mio-tflite] Revise to use new schema with mutable

This will revise to use new schema to support file size > 2G and with mutable methods. ONE-DCO-1.0-Signed-off-by: SaeHie Park <[email protected]>
Samsung · Aug 19, 2024 · d39036a · d39036a
1 parent 8f98b9a
commit d39036a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/compiler/mio-tflite2121/CMakeLists.txt b/compiler/mio-tflite2121/CMakeLists.txt
@@ -15,7 +15,9 @@ endif(NOT TensorFlowSource_FOUND)
 message(STATUS "Build mio-tflite2121: TRUE")
 message(STATUS "Build mio-tflite2121: with ${TensorFlowSource_DIR}")
 
-set(SCHEMA_FILE "${TensorFlowSource_DIR}/tensorflow/lite/schema/schema.fbs")
+# TODO rollback to use TensorFlowSource_DIR after upgrade
+#set(SCHEMA_FILE "${TensorFlowSource_DIR}/tensorflow/lite/schema/schema.fbs")
+set(SCHEMA_FILE "${NNAS_PROJECT_SOURCE_DIR}/res/TensorFlowLiteSchema/2.16.1/schema.fbs")
 
 # NOTE Use copy of schema.fbs as to provide unified way for circle also
 add_custom_command(OUTPUT "${CMAKE_CURRENT_BINARY_DIR}/schema.fbs"
@@ -24,7 +26,7 @@ add_custom_command(OUTPUT "${CMAKE_CURRENT_BINARY_DIR}/schema.fbs"
   DEPENDS "${SCHEMA_FILE}"
 )
 
-FlatBuffers_Target(mio_tflite2121
+FlatBuffersMuteable_Target(mio_tflite2121
   OUTPUT_DIR "${CMAKE_CURRENT_BINARY_DIR}/gen/mio/tflite"
   INCLUDE_DIR "${CMAKE_CURRENT_BINARY_DIR}/gen"
   SCHEMA_DIR "${CMAKE_CURRENT_BINARY_DIR}"