Commit 1f9c8668 authored by Paul's avatar Paul
Browse files

Get the input args

parent 7110eb0e
......@@ -635,7 +635,7 @@ struct find_mul_add_transpose_contiguous_reshaper_gemm
match::used_once(),
match::either_arg(0, 1)(match::is_constant().bind("c"), match::any().bind("x")));
return match::name("dot")(match::either_arg(0, 1)(
match_transpose_contiguous_reshaper(pw.bind("pointwise")), match::is_constant()));
match_transpose_contiguous_reshaper(match::args(pw.bind("pointwise"))), match::is_constant()));
}
void apply(module& m, const 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