diff --git a/tensorflow/lite/CMakeLists.txt b/tensorflow/lite/CMakeLists.txt index 5b5a3041584..74f1dc8253b 100644 --- a/tensorflow/lite/CMakeLists.txt +++ b/tensorflow/lite/CMakeLists.txt @@ -424,6 +424,9 @@ add_library(tensorflow::tensorflowlite ALIAS tensorflow-lite) # The benchmark tool. add_subdirectory(${TFLITE_SOURCE_DIR}/tools/benchmark) +# The label_image example. +add_subdirectory(${TFLITE_SOURCE_DIR}/examples/label_image) + # Python interpreter wrapper. add_library(_pywrap_tensorflow_interpreter_wrapper SHARED EXCLUDE_FROM_ALL ${TFLITE_SOURCE_DIR}/python/interpreter_wrapper/interpreter_wrapper.cc diff --git a/tensorflow/lite/examples/label_image/CMakeLists.txt b/tensorflow/lite/examples/label_image/CMakeLists.txt new file mode 100644 index 00000000000..c37ca78f1b0 --- /dev/null +++ b/tensorflow/lite/examples/label_image/CMakeLists.txt @@ -0,0 +1,68 @@ +# +# Copyright 2020 The TensorFlow Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# https://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + + +# The label_image example for Tensorflow Lite. + +populate_source_vars("${TFLITE_SOURCE_DIR}/examples/label_image" + TFLITE_LABEL_IMAGE_SRCS + FILTER "_test\\.cc$" +) +list(APPEND TFLITE_LABEL_IMAGE_SRCS + ${TF_SOURCE_DIR}/core/util/stats_calculator.cc + ${TFLITE_SOURCE_DIR}/profiling/memory_info.cc + ${TFLITE_SOURCE_DIR}/profiling/profile_summarizer.cc + ${TFLITE_SOURCE_DIR}/profiling/profile_summary_formatter.cc + ${TFLITE_SOURCE_DIR}/profiling/time.cc + ${TFLITE_SOURCE_DIR}/tools/command_line_flags.cc + ${TFLITE_SOURCE_DIR}/tools/delegates/default_execution_provider.cc + ${TFLITE_SOURCE_DIR}/tools/evaluation/utils.cc + ${TFLITE_SOURCE_DIR}/tools/tool_params.cc +) + +if(TFLITE_ENABLE_XNNPACK) + list(APPEND TFLITE_LABEL_IMAGE_SRCS + ${TFLITE_SOURCE_DIR}/tools/delegates/xnnpack_delegate_provider.cc + ) +else() + set(TFLITE_LABEL_IMAGE_CC_OPTIONS "-DTFLITE_WITHOUT_XNNPACK") +endif() # TFLITE_ENABLE_XNNPACK + +if(CMAKE_SYSTEM_NAME MATCHES "Android") + if(_TFLITE_ENABLE_NNAPI) + list(APPEND TFLITE_LABEL_IMAGE_SRCS + ${TFLITE_SOURCE_DIR}/tools/delegates/nnapi_delegate_provider.cc + ) + endif() # _TFLITE_ENABLE_NNAPI +endif() # Android + +if(TFLITE_ENABLE_GPU) + list(APPEND TFLITE_LABEL_IMAGE_SRCS + ${TFLITE_SOURCE_DIR}/tools/delegates/gpu_delegate_provider.cc + ) +endif() # TFLITE_ENABLE_GPU + +add_executable(label_image + EXCLUDE_FROM_ALL + ${TFLITE_LABEL_IMAGE_SRCS} +) +target_compile_options(label_image + PRIVATE + ${TFLITE_LABEL_IMAGE_CC_OPTIONS} +) +target_link_libraries(label_image + tensorflow-lite + ${CMAKE_DL_LIBS} +)