diff options
| -rw-r--r-- | delegate/cmake/Modules/FindTensorflow.cmake | 2 | ||||
| -rw-r--r-- | delegate/cmake/Modules/FindTfLite.cmake | 2 |
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 |