saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-19 19:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Enable Unsqueeze elimination (#1104)

Browse source
This commit is contained in:
stevenlix 2019-05-24 16:15:42 -07:00 committed by GitHub
parent ee6217972b
commit 2d92b95729
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/core/optimizer/graph_transformer_utils.cc
View file

@ -28,6 +28,7 @@ std::vector<std::unique_ptr<RewriteRule>> GenerateRewriteRules(TransformerLevel
case TransformerLevel::Level1:
rules.push_back(std::make_unique<EliminateIdentity>());
rules.push_back(std::make_unique<EliminateSlice>());
rules.push_back(std::make_unique<UnsqueezeElimination>());
rules.push_back(std::make_unique<EliminateDropout>());
break;