Commit 5a068d7e authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent 95665ff2
...@@ -50,8 +50,8 @@ struct layernorm_matcher ...@@ -50,8 +50,8 @@ struct layernorm_matcher
{ {
return f("div")(arg(0)(x_minus_mean()), return f("div")(arg(0)(x_minus_mean()),
arg(1)(skip_broadcasts(f("sqrt")(arg(0)(f("add")( arg(1)(skip_broadcasts(f("sqrt")(arg(0)(
either_arg(0, 1)(variance(), is_constant().bind("eps")))))))); f("add")(either_arg(0, 1)(variance(), is_constant().bind("eps"))))))));
} }
auto matcher() const { return layernorm_onnx(); } auto matcher() const { return layernorm_onnx(); }
......
...@@ -38,8 +38,7 @@ struct layernorm_base ...@@ -38,8 +38,7 @@ struct layernorm_base
template <class Self, class F> template <class Self, class F>
static auto reflect(Self& self, F f) static auto reflect(Self& self, F f)
{ {
return pack( return pack(f(self.epsilon, "epsilon"));
f(self.epsilon, "epsilon"));
} }
shape compute_shape(std::vector<shape> inputs, std::vector<module_ref> mods) const shape compute_shape(std::vector<shape> inputs, std::vector<module_ref> mods) 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