diff --git a/llvm/CMakeLists.txt b/llvm/CMakeLists.txt index 8f4d89749004b0..37218cd1093b10 100644 --- a/llvm/CMakeLists.txt +++ b/llvm/CMakeLists.txt @@ -803,7 +803,7 @@ if (TENSORFLOW_C_LIB_PATH) endif() string(REGEX MATCH "Location: ([^\n]*\n)" TF_PIP_LOC "${TF_PIP_OUT}") string(REPLACE "Location: " "" TF_PIP ${TF_PIP_LOC}) - set(TF_PROTO_HEADERS ${TF_PIP}/include) + set(TF_PROTO_HEADERS ${TF_PIP}/tensorflow/include) endif() include_directories(${TF_PROTO_HEADERS}) add_definitions("-DGOOGLE_PROTOBUF_NO_RTTI")