From d39036a0cc900d51cc39e80fb8a1830b60feec13 Mon Sep 17 00:00:00 2001 From: SaeHie Park Date: Mon, 19 Aug 2024 05:49:05 +0000 Subject: [PATCH] [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 --- compiler/mio-tflite2121/CMakeLists.txt | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/compiler/mio-tflite2121/CMakeLists.txt b/compiler/mio-tflite2121/CMakeLists.txt index 371118be83c..3cd9662d460 100644 --- 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}"