Browse Source

!5684 Fix some tiny bugs for converter in windows enviroment

Merge pull request !5684 from liuwenhao/master
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
541e65ac3b
1 changed files with 9 additions and 3 deletions
  1. +9
    -3
      mindspore/lite/tools/converter/parser/tflite/CMakeLists.txt

+ 9
- 3
mindspore/lite/tools/converter/parser/tflite/CMakeLists.txt View File

@@ -1,9 +1,15 @@
file(GLOB_RECURSE TFLITE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
*.cc
)
ADD_DEFINITIONS(-DFLATBUFFERS_LOCALE_INDEPENDENT=1)
find_library(FLATBUFFERS_LIBRARY flatbuffers HINTS ${TOP_DIR}/third_party/flatbuffers/build)
if (NOT WIN32)
ADD_DEFINITIONS(-DFLATBUFFERS_LOCALE_INDEPENDENT=1)
find_library(FLATBUFFERS_LIBRARY flatbuffers HINTS ${TOP_DIR}/third_party/flatbuffers/build)
endif ()
add_library(tflite_parser_mid OBJECT
${TFLITE_SRC_LIST}
)
target_link_libraries(tflite_parser_mid ${FLATBUFFERS_LIBRARY})
if (WIN32)
target_link_libraries(tflite_parser_mid mindspore::flatbuffers)
else ()
target_link_libraries(tflite_parser_mid ${FLATBUFFERS_LIBRARY})
endif ()

Loading…
Cancel
Save