Commit c2c5a632 authored by turneram's avatar turneram
Browse files

Formatting

parent cab8156e
...@@ -855,19 +855,22 @@ struct find_gelu_erf ...@@ -855,19 +855,22 @@ struct find_gelu_erf
{ {
static auto match_mul1() static auto match_mul1()
{ {
return match::name("mul")(args(match::any().bind("x"), match::skip_broadcasts(match::name("recip")))); return match::name("mul")(
args(match::any().bind("x"), match::skip_broadcasts(match::name("recip"))));
} }
static auto match_erf() { return match::name("erf")(match::arg(0)(match_mul1())); } static auto match_erf() { return match::name("erf")(match::arg(0)(match_mul1())); }
static auto match_add2() static auto match_add2()
{ {
return match::name("add")(args(match_erf(), match::skip_broadcasts(match::has_value(1.0f)))); return match::name("add")(
args(match_erf(), match::skip_broadcasts(match::has_value(1.0f))));
} }
static auto match_add1() static auto match_add1()
{ {
return match::name("add")(args(match::skip_broadcasts(match::is_constant()), match::name("dot"))); return match::name("add")(
args(match::skip_broadcasts(match::is_constant()), match::name("dot")));
} }
static auto match_mul2() { return match::name("mul")(match::args(match_add1(), match_add2())); } static auto match_mul2() { return match::name("mul")(match::args(match_add1(), match_add2())); }
......
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