Browse Source

handle initializer for input

pull/228/head
y00500818 5 years ago
parent
commit
55b74a0f3b
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      parser/onnx/onnx_parser.cc

+ 2
- 2
parser/onnx/onnx_parser.cc View File

@@ -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;
}



Loading…
Cancel
Save