Commit eeb92b57 authored by Paul's avatar Paul
Browse files

Formatting

parent c7b4fadd
......@@ -27,8 +27,7 @@ struct find_mul_conv
{
auto matcher() const
{
return match::name("mul")(match::either_arg(0, 1)(
conv_const_weights().bind("conv"),
return match::name("mul")(match::either_arg(0, 1)(conv_const_weights().bind("conv"),
match::name("broadcast").bind("a")));
}
......@@ -57,9 +56,10 @@ struct find_mul_add
auto matcher() const
{
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::is_constant().bind("a")
));
match::name("add")(match::either_arg(0, 1)(
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
......
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