diff options
Diffstat (limited to 'delegate/cmake/Modules/FindTensorflow.cmake')
-rw-r--r-- | delegate/cmake/Modules/FindTensorflow.cmake | 2 |
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) |