From cc2e198455d6e30791043ec9cc79107e6881daef Mon Sep 17 00:00:00 2001 From: xuanyue Date: Tue, 28 Sep 2021 14:04:53 +0800 Subject: [PATCH] fix the bug of setting model's output name --- mindspore/lite/tools/converter/CMakeLists.txt | 1 + .../{registry/parser_context.cc => converter_context.cc} | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) rename mindspore/lite/tools/converter/{registry/parser_context.cc => converter_context.cc} (100%) diff --git a/mindspore/lite/tools/converter/CMakeLists.txt b/mindspore/lite/tools/converter/CMakeLists.txt index 3ef8e6724ca..77fe534f389 100644 --- a/mindspore/lite/tools/converter/CMakeLists.txt +++ b/mindspore/lite/tools/converter/CMakeLists.txt @@ -17,6 +17,7 @@ endif() file(GLOB_RECURSE CONVERTER_SRC RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} ${CMAKE_CURRENT_SOURCE_DIR}/ops/*.cc ${CMAKE_CURRENT_SOURCE_DIR}/converter.cc + ${CMAKE_CURRENT_SOURCE_DIR}/converter_context.cc ${CMAKE_CURRENT_SOURCE_DIR}/converter_flags.cc ${CMAKE_CURRENT_SOURCE_DIR}/anf_transform.cc ${CMAKE_CURRENT_SOURCE_DIR}/graphdef_transform.cc diff --git a/mindspore/lite/tools/converter/registry/parser_context.cc b/mindspore/lite/tools/converter/converter_context.cc similarity index 100% rename from mindspore/lite/tools/converter/registry/parser_context.cc rename to mindspore/lite/tools/converter/converter_context.cc index 5b02155172f..21b8789913c 100644 --- a/mindspore/lite/tools/converter/registry/parser_context.cc +++ b/mindspore/lite/tools/converter/converter_context.cc @@ -14,10 +14,10 @@ * limitations under the License. */ -#include "include/registry/parser_context.h" +#include "tools/converter/converter_context.h" #include #include -#include "tools/converter/converter_context.h" +#include "include/registry/parser_context.h" namespace mindspore { namespace converter {