GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-29 04:17:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix onnx normalize expand ghost shape

Browse Source
This commit is contained in:
nihuini
2021-03-03 17:21:45 +08:00
parent c910574b5b
commit f7cbcaa72b
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/onnx/onnx2ncnn.cpp
+4
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));