diff --git a/mindspore/lite/tools/converter/parser/caffe/CMakeLists.txt b/mindspore/lite/tools/converter/parser/caffe/CMakeLists.txt index 97406e5bc7..4562c24290 100644 --- 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