Browse Source

fix onnx normalize expand ghost shape

tags/20210322
nihuini 5 years ago
parent
commit
f7cbcaa72b
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      tools/onnx/onnx2ncnn.cpp

+ 4
- 0
tools/onnx/onnx2ncnn.cpp View File

@@ -1204,6 +1204,10 @@ static void fuse_normalize(onnx::GraphProto* mutable_graph, std::map<std::string
node_reference[node_shape->output(0)] -= 1;
}
node_reference[node3->output(0)] -= 1;
if (node3->input_size() == 2)
{
node_reference[node3->input(1)] -= 1;
}

blob_names.erase(node->output(0));
blob_names.erase(node2->output(0));


Loading…
Cancel
Save