Commit eeb92b57 authored by Paul's avatar Paul
Browse files

Formatting

parent c7b4fadd
...@@ -27,8 +27,7 @@ struct find_mul_conv ...@@ -27,8 +27,7 @@ struct find_mul_conv
{ {
auto matcher() const auto matcher() const
{ {
return match::name("mul")(match::either_arg(0, 1)( return match::name("mul")(match::either_arg(0, 1)(conv_const_weights().bind("conv"),
conv_const_weights().bind("conv"),
match::name("broadcast").bind("a"))); match::name("broadcast").bind("a")));
} }
...@@ -57,9 +56,10 @@ struct find_mul_add ...@@ -57,9 +56,10 @@ struct find_mul_add
auto matcher() const auto matcher() const
{ {
return match::name("mul")(match::either_arg(0, 1)( return match::name("mul")(match::either_arg(0, 1)(
match::name("add")(match::either_arg(0, 1)(match::any().bind("x"), match::any_of(conv_const_weights(), match::is_constant()).bind("y"))), match::name("add")(match::either_arg(0, 1)(
match::is_constant().bind("a") match::any().bind("x"),
)); match::any_of(conv_const_weights(), match::is_constant()).bind("y"))),
match::is_constant().bind("a")));
} }
void apply(program& p, match::matcher_result r) const void apply(program& p, match::matcher_result r) const
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment