tensorflow-lite.mk 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. ################################################################################
  2. #
  3. # tensorflow-lite
  4. #
  5. ################################################################################
  6. TENSORFLOW_LITE_VERSION = 2.11.0
  7. TENSORFLOW_LITE_SITE = $(call github,tensorflow,tensorflow,v$(TENSORFLOW_LITE_VERSION))
  8. TENSORFLOW_LITE_INSTALL_STAGING = YES
  9. TENSORFLOW_LITE_LICENSE = Apache-2.0
  10. TENSORFLOW_LITE_LICENSE_FILES = LICENSE
  11. TENSORFLOW_LITE_SUBDIR = tensorflow/lite
  12. TENSORFLOW_LITE_SUPPORTS_IN_SOURCE_BUILD = NO
  13. TENSORFLOW_LITE_DEPENDENCIES += \
  14. host-pkgconf \
  15. host-flatbuffers \
  16. cpuinfo \
  17. eigen \
  18. farmhash \
  19. fft2d \
  20. flatbuffers \
  21. gemmlowp \
  22. libabseil-cpp \
  23. neon-2-sse
  24. TENSORFLOW_LITE_CONF_OPTS = \
  25. -Dabsl_DIR=$(STAGING_DIR)/usr/lib/cmake/absl \
  26. -DBUILD_SHARED_LIBS=ON \
  27. -DCMAKE_CXX_FLAGS="$(TARGET_CXXFLAGS) -I$(STAGING_DIR)/usr/include/gemmlowp" \
  28. -DCMAKE_FIND_PACKAGE_PREFER_CONFIG=ON \
  29. -DCMAKE_POSITION_INDEPENDENT_CODE=ON \
  30. -DEigen3_DIR=$(STAGING_DIR)/usr/share/eigen3/cmake \
  31. -DFETCHCONTENT_FULLY_DISCONNECTED=ON \
  32. -DFETCHCONTENT_QUIET=OFF \
  33. -DFFT2D_SOURCE_DIR=$(STAGING_DIR)/usr/include/fft2d \
  34. -DFlatBuffers_DIR=$(STAGING_DIR)/usr/lib/cmake/flatbuffers \
  35. -DNEON_2_SSE_DIR=$(STAGING_DIR)/usr/lib/cmake/NEON_2_SSE \
  36. -DSYSTEM_FARMHASH=ON \
  37. -DTFLITE_ENABLE_EXTERNAL_DELEGATE=ON \
  38. -DTFLITE_ENABLE_GPU=OFF \
  39. -DTFLITE_ENABLE_INSTALL=ON \
  40. -DTFLITE_ENABLE_MMAP=ON \
  41. -DTFLITE_ENABLE_NNAPI=OFF
  42. ifeq ($(BR2_PACKAGE_RUY),y)
  43. TENSORFLOW_LITE_DEPENDENCIES += ruy
  44. TENSORFLOW_LITE_CONF_OPTS += -DTFLITE_ENABLE_RUY=ON
  45. else
  46. TENSORFLOW_LITE_CONF_OPTS += -DTFLITE_ENABLE_RUY=OFF
  47. endif
  48. ifeq ($(BR2_PACKAGE_XNNPACK),y)
  49. TENSORFLOW_LITE_DEPENDENCIES += xnnpack
  50. TENSORFLOW_LITE_CONF_OPTS += -DTFLITE_ENABLE_XNNPACK=ON -Dxnnpack_POPULATED=ON
  51. else
  52. TENSORFLOW_LITE_CONF_OPTS += -DTFLITE_ENABLE_XNNPACK=OFF
  53. endif
  54. $(eval $(cmake-package))