From 31fcdecaee469c94aa86b1972ed4c85c3f1ebb9f Mon Sep 17 00:00:00 2001 From: daquexian Date: Mon, 26 Jun 2023 16:39:51 +0800 Subject: [PATCH 1/2] bump to 0.3.17 Signed-off-by: daquexian Signed-off-by: AlexHJH --- VERSION_NUMBER | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/VERSION_NUMBER b/VERSION_NUMBER index ec96a6218..e5a9958c3 100644 --- a/VERSION_NUMBER +++ b/VERSION_NUMBER @@ -1 +1 @@ -0.3.16 +0.3.17 From 36d561fc5cc776b97e112add0cf5ea223120e475 Mon Sep 17 00:00:00 2001 From: AlexHJH Date: Wed, 2 Aug 2023 00:21:23 +0800 Subject: [PATCH 2/2] fix missing functions Signed-off-by: AlexHJH --- onnxoptimizer/optimize.h | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/onnxoptimizer/optimize.h b/onnxoptimizer/optimize.h index f25e6c3f6..9404b8761 100644 --- a/onnxoptimizer/optimize.h +++ b/onnxoptimizer/optimize.h @@ -46,12 +46,23 @@ struct Optimizer { ModelProto mp_out = PrepareOutput(mp_in); this->pass_manager->run(*g); ExportModelProto(&mp_out, g); + + // Maybe we can optimize these functions, now just copy + AddFunctionsToModel(mp_in, mp_out); return mp_out; } private: std::shared_ptr pass_manager; + void AddFunctionsToModel(const ModelProto &original_model, + ModelProto &output_model) { + for (const auto& function_proto : original_model.functions()) { + auto* p_f = output_model.add_functions(); + p_f->CopyFrom(function_proto); + } + } + ModelProto AddInitializerToInput(const ModelProto &original_model) { ModelProto model = original_model; std::vector input_names;