diff options
| author | 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com> | 2019-03-18 10:29:09 +0900 |
|---|---|---|
| committer | 오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com> | 2019-03-18 10:29:09 +0900 |
| commit | 452fc784d1cb0c75240e9d986c985b6ddaa7ea18 (patch) | |
| tree | 7a9a647c46c497c03cf936c2329cf2c2a77830d7 /externals/CMakeLists.txt | |
| parent | 105270fb1f6e6701a0eb56fde84941edbedf1000 (diff) | |
| download | nnfw-452fc784d1cb0c75240e9d986c985b6ddaa7ea18.tar.gz nnfw-452fc784d1cb0c75240e9d986c985b6ddaa7ea18.tar.bz2 nnfw-452fc784d1cb0c75240e9d986c985b6ddaa7ea18.zip | |
Do NOT use CMAKE_CURRENT_SOURCE_DIR for tflite build (#4761)
The current tflite build script uses the hardcoded location of TensorFlow
source code.
This commit removes such hardcoded location and uses TensorFlowSource_DIR
variable instead.
Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
Diffstat (limited to 'externals/CMakeLists.txt')
| -rw-r--r-- | externals/CMakeLists.txt | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/externals/CMakeLists.txt b/externals/CMakeLists.txt index 4ef3d2170..ce6d3cb16 100644 --- a/externals/CMakeLists.txt +++ b/externals/CMakeLists.txt @@ -3,9 +3,6 @@ if(NOT BUILD_TENSORFLOW_LITE) return() endif(NOT BUILD_TENSORFLOW_LITE) -set(TENSORFLOW_BASE ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow) -set(TENSORFLOW_LITE_BASE ${TENSORFLOW_BASE}/tensorflow/contrib/lite) - # Required source & package nnfw_find_package(AbslSource REQUIRED) nnfw_find_package(Eigen REQUIRED) @@ -14,6 +11,8 @@ nnfw_find_package(FlatBuffersSource REQUIRED) nnfw_find_package(GEMMLowpSource REQUIRED) nnfw_find_package(TensorFlowSource REQUIRED) +set(TENSORFLOW_LITE_BASE ${TensorFlowSource_DIR}/tensorflow/contrib/lite) + # # Tensorflow Lite library # @@ -49,7 +48,7 @@ list(APPEND TFLITE_SRCS ${TFLITE_PROFILING_SRCS}) list(APPEND TFLITE_SRCS "${FarmhashSource_DIR}/src/farmhash.cc") -list(APPEND TFLITE_INCLUDES "${CMAKE_CURRENT_SOURCE_DIR}/tensorflow") +list(APPEND TFLITE_INCLUDES "${TensorFlowSource_DIR}") list(APPEND TFLITE_INCLUDES "${AbslSource_DIR}") list(APPEND TFLITE_INCLUDES "${GEMMLowpSource_DIR}") list(APPEND TFLITE_INCLUDES "${FarmhashSource_DIR}/src") |