infra/cmake/packages/TensorFlowLite-1.7/TensorFlowLiteConfig.cmake

   1 function(_TensorFlowLite_import)
   2   nnas_find_package(TensorFlowSource EXACT 1.7 QUIET)
   3
   4   if(NOT TensorFlowSource_FOUND)
   5     set(TensorFlowLite_FOUND FALSE PARENT_SCOPE)
   6     return()
   7   endif(NOT TensorFlowSource_FOUND)
   8
   9   # TensorFlow 1.7 downloads FlatBuffers from https://github.com/google/flatbuffers/archive/971a68110e4.tar.gz
  10   #
  11   # FlatBuffers 1.8 is compatible with 971a68110e4.
  12   nnas_find_package(FlatBuffersSource EXACT 1.8 QUIET)
  13
  14   if(NOT FlatBuffersSource_FOUND)
  15     set(TensorFlowLite_FOUND FALSE PARENT_SCOPE)
  16     return()
  17   endif(NOT FlatBuffersSource_FOUND)
  18
  19   nnas_find_package(Farmhash QUIET)
  20
  21   if(NOT Farmhash_FOUND)
  22     set(TensorFlowLite_FOUND FALSE PARENT_SCOPE)
  23     return()
  24   endif(NOT Farmhash_FOUND)
  25
  26   nnas_find_package(Eigen QUIET)
  27
  28   if(NOT Eigen_FOUND)
  29     set(TensorFlowLite_FOUND FALSE PARENT_SCOPE)
  30     return()
  31   endif(NOT Eigen_FOUND)
  32
  33   nnas_find_package(GEMMLowp QUIET)
  34
  35   if(NOT GEMMLowp_FOUND)
  36     set(TensorFlowLite_FOUND FALSE PARENT_SCOPE)
  37     return()
  38   endif(NOT GEMMLowp_FOUND)
  39
  40   nnas_find_package(NEON2SSE QUIET)
  41
  42   if(NOT NEON2SSE_FOUND)
  43     set(TensorFlowLite_FOUND FALSE PARENT_SCOPE)
  44     return()
  45   endif(NOT NEON2SSE_FOUND)
  46
  47   if(NOT TARGET tensorflowlite-1.7)
  48     nnas_include(ExternalProjectTools)
  49     add_extdirectory("${CMAKE_CURRENT_LIST_DIR}/Lite" tflite-1.7)
  50   endif(NOT TARGET tensorflowlite-1.7)
  51
  52   set(TensorFlowLite_FOUND TRUE PARENT_SCOPE)
  53 endfunction(_TensorFlowLite_import)
  54
  55 _TensorFlowLite_import()