commit b55f8e848a841e4d75fce0e8324c23c3876d2f71
author Michalis Spyrou <michalis.spyrou@arm.com> Thu Jul 22 11:23:11 2021 +0100
committer Georgios Pinitas <georgios.pinitas@arm.com> Thu Jul 29 12:11:10 2021 +0000
tree e3e283f3be1d7c776e2cff80f26ba0e4f30c69eb
parent 8e2f64f214fa3ce5834db966222fa3b804e236a2

Port NEConvolutionLayer

Resolves: COMPMID-4507

Change-Id: I9557026ec0052b5585994f7a1300a14565c976d0
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5964
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>

src/runtime/cpu/operators/CpuGemmConvolution.cpp

diff --git a/src/runtime/cpu/operators/CpuGemmConvolution.cpp b/src/runtime/cpu/operators/CpuGemmConvolution.cpp
index 6a78b0c..864d7e2 100644
--- a/src/runtime/cpu/operators/CpuGemmConvolution.cpp
+++ b/src/runtime/cpu/operators/CpuGemmConvolution.cpp
@@ -575,8 +575,7 @@
     if(!_is_prepared)
     {
         // Run weights reshaping and mark original weights tensor as unused
-        ITensor            *weights_reshaped_p = utils::cast::polymorphic_downcast<ITensor *>(tensors.get_tensor(offset_int_vec(WeightsReshaped)));
-        CpuAuxTensorHandler weights_reshaped(_weights_reshaped, *weights_reshaped_p);
+        CpuAuxTensorHandler weights_reshaped(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors);
         auto                weights = tensors.get_const_tensor(TensorType::ACL_SRC_1);
         ITensorPack         pack =
         {