Skip to content

Commit 25b5ada

Browse files
author
Han Wang
committed
fix bug in finding protobuf
1 parent 708370b commit 25b5ada

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

source/cmake/Findtensorflow.cmake

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ find_path(TensorFlow_INCLUDE_DIRS
3434
)
3535
find_path(TensorFlow_INCLUDE_DIRS_GOOGLE
3636
NAMES
37-
google/protobuf/port.h
37+
google/protobuf/type.pb.h
3838
PATHS ${TensorFlow_search_PATHS}
3939
PATH_SUFFIXES "/include"
4040
NO_DEFAULT_PATH
@@ -43,7 +43,7 @@ list(APPEND TensorFlow_INCLUDE_DIRS ${TensorFlow_INCLUDE_DIRS_GOOGLE})
4343

4444
if (NOT TensorFlow_INCLUDE_DIRS AND tensorflow_FIND_REQUIRED)
4545
message(FATAL_ERROR
46-
"Not found 'include/tensorflow/core/public/session.h' directory in path '${TensorFlow_search_PATHS}' "
46+
"Not found 'tensorflow/core/public/session.h' directory in path '${TensorFlow_search_PATHS}' "
4747
"You can manually set the tensorflow install path by -DTENSORFLOW_ROOT ")
4848
endif ()
4949

0 commit comments

Comments
 (0)