Skip to content

GitLab

Commit e01c70e6 authored by Paul's avatar Paul
Browse files

Add assert for negative numbers

parent 2401848f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/targets/cpu/cpu_lowering.cpp
View file @ e01c70e6
...@@ -65,6 +65,7 @@ struct cpu_batch_norm_inference ...@@ -65,6 +65,7 @@ struct cpu_batch_norm_inference
dfor(num_batch, num_channels, image_height, image_width)( dfor(num_batch, num_channels, image_height, image_width)(
[&](std::size_t n, std::size_t c, std::size_t h, std::size_t w) { [&](std::size_t n, std::size_t c, std::size_t h, std::size_t w) {
assert((variance(c) + epsilon) > 0);
result(n, c, h, w) = gamma(c) * (buffer(n, c, h, w) - mean(c)) / result(n, c, h, w) = gamma(c) * (buffer(n, c, h, w) - mean(c)) /
std::sqrt(variance(c) + epsilon) + std::sqrt(variance(c) + epsilon) +
bias(c); bias(c);
...@@ -79,6 +80,7 @@ struct cpu_batch_norm_inference ...@@ -79,6 +80,7 @@ struct cpu_batch_norm_inference
dfor(num_batch, num_channels, image_height, image_width)( dfor(num_batch, num_channels, image_height, image_width)(
[&](std::size_t n, std::size_t c, std::size_t h, std::size_t w) { [&](std::size_t n, std::size_t c, std::size_t h, std::size_t w) {
assert((variance(c, h, w) + epsilon) > 0);
result(n, c, h, w) = gamma(c, h, w) * result(n, c, h, w) = gamma(c, h, w) *
(buffer(n, c, h, w) - mean(c, h, w)) / (buffer(n, c, h, w) - mean(c, h, w)) /
std::sqrt(variance(c, h, w) + epsilon) + std::sqrt(variance(c, h, w) + 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