From 1b2ee3eb6991efcbd09cde84ab0c383f15648738 Mon Sep 17 00:00:00 2001
From: David Mansell <David.Mansell@arm.com>
Date: Tue, 22 Aug 2023 13:27:03 +0100
Subject: CPU: Depthwise: Generate correct size for input indirection array.

Signed-off-by: David Mansell <David.Mansell@arm.com>
Change-Id: I359ed0703f4036e017b34b622f76b630cefac973
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10183
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
---
 .../NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/core/NEON/kernels')

diff --git a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp
index ca5026b6e0..e2d05560a1 100644
--- a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp
+++ b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp
@@ -186,7 +186,7 @@ class GenericInputArrayElement
   static size_t get_element_size(const WorkspaceArgs<IDepthfirstStrategy, OutputStage> &args)
   {
     const auto kernel_points = args.depthwise_args.kernel_rows * args.depthwise_args.kernel_cols;
-    return sizeof(T **) * args.strategy->get_input_rows() * args.strategy->get_input_cols() * kernel_points;
+    return sizeof(T **) * args.strategy->get_output_rows() * args.strategy->get_output_cols() * kernel_points;
   }
 
   template <class WorkspaceType, class OutputStage>
-- 
cgit v1.2.1