!5684 Fix some tiny bugs for converter in windows enviroment
Merge pull request !5684 from liuwenhao/masterpull/5684/MERGE
commit
541e65ac3b
@ -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…
Reference in new issue