Commit 3987066f authored by Paul's avatar Paul
Browse files

Formatting

parent b889693c
......@@ -34,20 +34,11 @@ void fwd_conv_batchnorm_rewrite::apply(program& p) const
argument a{s};
argument b{s};
visit_all(gamma, bias, mean, variance, a, b)(
[&](auto gamma2,
auto bias2,
auto mean2,
auto variance2,
auto a2,
auto b2) {
[&](auto gamma2, auto bias2, auto mean2, auto variance2, auto a2, auto b2) {
dfor(a.get_shape().elements())(
[&](std::size_t c) {
a2[c] =
gamma2[c] / std::sqrt(variance2[c] + epsilon);
});
[&](std::size_t c) { a2[c] = gamma2[c] / std::sqrt(variance2[c] + epsilon); });
dfor(b.get_shape().elements())([&](std::size_t c) {
b2[c] =
bias2[c] - (gamma2[c] * mean2[c] / std::sqrt(variance2[c] + epsilon));
b2[c] = bias2[c] - (gamma2[c] * mean2[c] / std::sqrt(variance2[c] + epsilon));
});
});
......
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