26 lines
595 B
CMake
26 lines
595 B
CMake
include(FindPackageHandleStandardArgs)
|
|
|
|
unset(TENSORFLOW_FOUND)
|
|
|
|
find_path(TensorFlow_INCLUDE_DIR
|
|
NAMES
|
|
tensorflow/core
|
|
tensorflow/cc
|
|
third_party
|
|
HINTS
|
|
/usr/include/
|
|
/usr/local/include/)
|
|
|
|
find_library(TensorFlow_LIBRARY NAMES tensorflow_all
|
|
HINTS
|
|
/usr/lib
|
|
/usr/local/lib)
|
|
|
|
find_package_handle_standard_args(TensorFlow DEFAULT_MSG TensorFlow_INCLUDE_DIR TensorFlow_LIBRARY)
|
|
|
|
if(TENSORFLOW_FOUND)
|
|
set(TensorFlow_LIBRARIES ${TensorFlow_LIBRARY})
|
|
set(TensorFlow_INCLUDE_DIRS ${TensorFlow_INCLUDE_DIR})
|
|
endif()
|
|
|