# /path/to/tensorflow: tensorflow root path when use tflite library
set(USE_TENSORFLOW_PATH none)
+# Required for full builds with TFLite. Not needed for runtime with TFLite.
+# /path/to/flatbuffers: flatbuffers root path when using tflite library
+set(USE_FLATBUFFERS_PATH none)
+
# Possible values:
# - OFF: disable tflite support for edgetpu
# - /path/to/edgetpu: use specific path to edgetpu library
if(NOT USE_TFLITE STREQUAL "OFF")
message(STATUS "Build with contrib.tflite")
- if (USE_TENSORFLOW_PATH STREQUAL "none")
+ if (USE_TENSORFLOW_PATH STREQUAL "none")
set(USE_TENSORFLOW_PATH ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow)
endif()
find_library(TFLITE_CONTRIB_LIB libtensorflow-lite.a ${USE_TFLITE})
list(APPEND TVM_RUNTIME_LINKER_LIBS ${TFLITE_CONTRIB_LIB})
- list(APPEND TVM_RUNTIME_LINKER_LIBS rt dl flatbuffers)
+
+ if (NOT USE_FLATBUFFERS_PATH STREQUAL "none")
+ include_directories(${USE_FLATBUFFERS_PATH}/include)
+ endif()
endif()