aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJan Eilers <jan.eilers@arm.com>2020-11-20 10:57:51 +0000
committerFrancis Murtagh <francis.murtagh@arm.com>2020-11-23 16:26:40 +0000
commit2ab0c44f4056e5326a1311aae3c22331813ebbc0 (patch)
treeb79443726decd55d5a1b7880951cf0eeafd7968f
parent199ab1a39fa45849a74853745e90fa8cd5e5ea6e (diff)
downloadarmnn-2ab0c44f4056e5326a1311aae3c22331813ebbc0.tar.gz
IVGCVSW-5574 Change directory for tensorflow libs for the delegate
* TENSORFLOW_ROOT is already in use with a different expected value Signed-off-by: Jan Eilers <jan.eilers@arm.com> Change-Id: I199987989678d8712a1f3e0a79ae066376429eec
-rw-r--r--delegate/cmake/Modules/FindTensorflow.cmake2
-rw-r--r--delegate/cmake/Modules/FindTfLite.cmake2
2 files changed, 2 insertions, 2 deletions
diff --git a/delegate/cmake/Modules/FindTensorflow.cmake b/delegate/cmake/Modules/FindTensorflow.cmake
index 8b47d30362..8f90011a65 100644
--- a/delegate/cmake/Modules/FindTensorflow.cmake
+++ b/delegate/cmake/Modules/FindTensorflow.cmake
@@ -18,7 +18,7 @@ find_library(Tensorflow_LIB
NAMES
tensorflow_all
HINTS
- ${TENSORFLOW_ROOT})
+ ${TENSORFLOW_LIB_DIR})
## Set TENSORFLOW_FOUND
find_package_handle_standard_args(Tensorflow DEFAULT_MSG Tensorflow_INCLUDE_DIR Tensorflow_LIB)
diff --git a/delegate/cmake/Modules/FindTfLite.cmake b/delegate/cmake/Modules/FindTfLite.cmake
index 96e15db108..9bb117efac 100644
--- a/delegate/cmake/Modules/FindTfLite.cmake
+++ b/delegate/cmake/Modules/FindTfLite.cmake
@@ -11,7 +11,7 @@ find_path(TfLite_INCLUDE_DIR
tensorflow/lite
third_party
HINTS
- ${TENSORFLOW_ROOT}/..)
+ ${TENSORFLOW_ROOT})
find_library(TfLite_LIB
NAMES