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.cmake30
1 files changed, 30 insertions, 0 deletions
diff --git a/delegate/cmake/Modules/FindTensorflow.cmake b/delegate/cmake/Modules/FindTensorflow.cmake
new file mode 100644
index 0000000000..8b47d30362
--- /dev/null
+++ b/delegate/cmake/Modules/FindTensorflow.cmake
@@ -0,0 +1,30 @@
+#
+# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
+# SPDX-License-Identifier: MIT
+#
+
+include(FindPackageHandleStandardArgs)
+unset(TENSORFLOW_FOUND)
+
+find_path(Tensorflow_INCLUDE_DIR
+ NAMES
+ tensorflow/core
+ tensorflow/cc
+ third_party
+ HINTS
+ ${TENSORFLOW_ROOT})
+
+find_library(Tensorflow_LIB
+ NAMES
+ tensorflow_all
+ HINTS
+ ${TENSORFLOW_ROOT})
+
+## Set TENSORFLOW_FOUND
+find_package_handle_standard_args(Tensorflow DEFAULT_MSG Tensorflow_INCLUDE_DIR Tensorflow_LIB)
+
+## Set external variables for usage in CMakeLists.txt
+if(TENSORFLOW_FOUND)
+ set(Tensorflow_LIB ${Tensorflow_LIB})
+ set(Tensorflow_INCLUDE_DIRS ${Tensorflow_INCLUDE_DIR})
+endif() \ No newline at end of file