1# 2# Copyright © 2020 Arm Ltd and Contributors. All rights reserved. 3# SPDX-License-Identifier: MIT 4# 5 6include(FindPackageHandleStandardArgs) 7unset(TFLITE_FOUND) 8 9find_path(TfLite_INCLUDE_DIR 10 NAMES 11 tensorflow/lite 12 third_party 13 HINTS 14 ${TENSORFLOW_ROOT}) 15 16find_library(TfLite_LIB 17 NAMES 18 "libtensorflow_lite_all.so" 19 HINTS 20 ${TFLITE_LIB_ROOT}) 21 22find_path(TfLite_Schema_INCLUDE_PATH 23 schema_generated.h 24 HINTS 25 ${TFLITE_LIB_ROOT}/tensorflow/lite/schema) 26 27## Set TFLITE_FOUND 28find_package_handle_standard_args(TfLite DEFAULT_MSG TfLite_INCLUDE_DIR TfLite_LIB TfLite_Schema_INCLUDE_PATH) 29 30## Set external variables for usage in CMakeLists.txt 31if(TFLITE_FOUND) 32 set(TfLite_LIB ${TfLite_LIB}) 33 set(TfLite_INCLUDE_DIR ${TfLite_INCLUDE_DIR}) 34 set(TfLite_Schema_INCLUDE_PATH ${TfLite_Schema_INCLUDE_PATH}) 35endif()