wasm: bugfix in InstanceNormalization (#104)

microsoft · Mar 7, 2019 · 7429f07 · 7429f07
1 parent ca3e928
commit 7429f07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/wasm-ops/instance-normalization.cpp b/src/wasm-ops/instance-normalization.cpp
@@ -27,7 +27,7 @@ void instance_normalization_f32_imp(float *X, float *Y, int32_t batch_size,
   size_t iterEnd;
   size_t currentChannel;
 
-  for (size_t nc = 0; nc < batch_size * channel_size; nc++) {
+  for (size_t nc = 0; nc < batch_size * num_channels; nc++) {
     physicalOffset = nc * channel_size;
     iterEnd = physicalOffset + channel_size;
     currentChannel = nc % num_channels;