caffe converted flatten op is not comiled

5 years ago · f1f53acefd
parent 74bb5675be
commit f1f53acefd
1 changed files with 2 additions and 0 deletions
--- a/mindspore/lite/tools/converter/parser/caffe/CMakeLists.txt
+++ b/mindspore/lite/tools/converter/parser/caffe/CMakeLists.txt
@ -18,6 +18,8 @@ add_library(caffe_parser_mid OBJECT
        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_deconvolution_parser.h
        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_eltwise_parser.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_eltwise_parser.h
+        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_flatten_parser.cc
+        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_flatten_parser.h
        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_innerproduct_parser.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_innerproduct_parser.h
        ${CMAKE_CURRENT_SOURCE_DIR}/caffe_inspector.cc