From 55b74a0f3bac5d6d033aafc99ad36b6043d04bec Mon Sep 17 00:00:00 2001 From: y00500818 Date: Mon, 25 Jan 2021 14:16:43 +0800 Subject: [PATCH] handle initializer for input --- parser/onnx/onnx_parser.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc index b4ef845..d6dfcdb 100644 --- a/parser/onnx/onnx_parser.cc +++ b/parser/onnx/onnx_parser.cc @@ -161,10 +161,10 @@ Status OnnxModelParser::ParseInput(ge::onnx::GraphProto &onnx_graph, ge::onnx::ValueInfoProto value_info = onnx_graph.input(i); GELOGI("The index of %d input name : %s.", i, value_info.name().c_str()); - // The input are possibly initialized by a default value found in ‘initializer.’ + /// if the input is initialized by a default value found in ‘initializer’, + /// it will be considered as a const node. auto initializer_iter = initializer_name_tensor.find(value_info.name()); if (initializer_iter != initializer_name_tensor.end()) { - initializer_name_tensor.erase(initializer_iter); continue; }