Skip to content

Commit

Permalink
[circle2circle] Enable forward_reshape_to_unaryop option (Samsung#5871)
Browse files Browse the repository at this point in the history
This will enable forward_reshape_to_unaryop option.

ONE-DCO-1.0-Signed-off-by: SaeHie Park <[email protected]>
  • Loading branch information
seanshpark authored Jan 28, 2021
1 parent b87cc06 commit 940b24a
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions compiler/circle2circle/src/Circle2Circle.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,12 @@ int entry(int argc, char **argv)
.default_value(false)
.help("This will fold SparseToDense operator");

arser.add_argument("--forward_reshape_to_unaryop")
.nargs(0)
.required(false)
.default_value(false)
.help("This will move Reshape after UnaryOp for centain condition");

arser.add_argument("--fuse_activation_function")
.nargs(0)
.required(false)
Expand Down Expand Up @@ -323,6 +329,8 @@ int entry(int argc, char **argv)
options->enable(Algorithms::FoldDequantize);
if (arser.get<bool>("--fold_sparse_to_dense"))
options->enable(Algorithms::FoldSparseToDense);
if (arser.get<bool>("--forward_reshape_to_unaryop"))
options->enable(Algorithms::ForwardReshapeToUnaryOp);
if (arser.get<bool>("--fuse_activation_function"))
options->enable(Algorithms::FuseActivationFunction);
if (arser.get<bool>("--fuse_batchnorm_with_conv"))
Expand Down

0 comments on commit 940b24a

Please sign in to comment.