From bff26a32c4828297cf1742121cb19ee020575f5b Mon Sep 17 00:00:00 2001 From: chxin66 <57057788+chxin66@users.noreply.github.com> Date: Mon, 7 Aug 2023 13:20:35 +0800 Subject: [PATCH] fix size compute bug in lrn (#626) Signed-off-by: Chen Co-authored-by: Chen --- src/tim/vx/ops/localresponsenormalization_test.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tim/vx/ops/localresponsenormalization_test.cc b/src/tim/vx/ops/localresponsenormalization_test.cc index 16b7ff2..fd47d26 100644 --- a/src/tim/vx/ops/localresponsenormalization_test.cc +++ b/src/tim/vx/ops/localresponsenormalization_test.cc @@ -48,7 +48,7 @@ TEST(localresponsenormalization, axis_0_shape_6_1_1_1_float) { in_data.size() * sizeof(float))); int radius = 2; - int size = radius * 2; + int size = radius * 2 + 1; float alpha = 4.0, beta = 0.5, bias = 9.0; auto op = graph->CreateOperation( size, alpha, beta, bias, 0); @@ -88,7 +88,7 @@ TEST(localresponsenormalization, axis_1_shape_2_6_float) { in_data.size() * sizeof(float))); int radius = 2; - int size = radius * 2; + int size = radius * 2 + 1; float alpha = 4.0, beta = 0.5, bias = 9.0; auto op = graph->CreateOperation( size, alpha, beta, bias, 1);