diff options
Diffstat (limited to 'compiler/mir-interpreter/src/ops/AvgPool2D.cpp')
| -rw-r--r-- | compiler/mir-interpreter/src/ops/AvgPool2D.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/compiler/mir-interpreter/src/ops/AvgPool2D.cpp b/compiler/mir-interpreter/src/ops/AvgPool2D.cpp index 3f1d65100..3f74cd1e8 100644 --- a/compiler/mir-interpreter/src/ops/AvgPool2D.cpp +++ b/compiler/mir-interpreter/src/ops/AvgPool2D.cpp @@ -72,7 +72,7 @@ void AvgPool2DImpl<T>::run(const ops::AvgPool2DOp &op, const TensorVariant &inpu // Assuming NHWC format. for (int i = 0; i < num_spatial_dims; ++i) in_index.at(1 + i) = - out_index.at(1 + i) * strides[i] + window_index.at(i) - padding_before[i]; + out_index.at(1 + i) * strides[i] + window_index.at(i) - padding_before[i]; if (in_range.contains(in_index)) { @@ -145,7 +145,7 @@ void AvgPool2DImpl<uint8_t>::run(const ops::AvgPool2DOp &op, const TensorVariant // Assuming NHWC format. for (int i = 0; i < num_spatial_dims; ++i) in_index.at(1 + i) = - out_index.at(1 + i) * strides[i] + window_index.at(i) - padding_before[i]; + out_index.at(1 + i) * strides[i] + window_index.at(i) - padding_before[i]; if (in_range.contains(in_index)) { |