aboutsummaryrefslogtreecommitdiff
path: root/delegate/cmake/Modules/FindTensorflow.cmake
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/cmake/Modules/FindTensorflow.cmake')
-rw-r--r--delegate/cmake/Modules/FindTensorflow.cmake2
1 files changed, 1 insertions, 1 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)