diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-07-22 11:23:11 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-29 12:11:10 +0000 |
| commit | b55f8e848a841e4d75fce0e8324c23c3876d2f71 (patch) | |
| tree | e3e283f3be1d7c776e2cff80f26ba0e4f30c69eb /src/runtime/cpu/operators/CpuGemmConvolution.cpp | |
| parent | 8e2f64f214fa3ce5834db966222fa3b804e236a2 (diff) | |
| download | ComputeLibrary-b55f8e848a841e4d75fce0e8324c23c3876d2f71.tar.gz | |
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>
Diffstat (limited to 'src/runtime/cpu/operators/CpuGemmConvolution.cpp')
| -rw-r--r-- | src/runtime/cpu/operators/CpuGemmConvolution.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/runtime/cpu/operators/CpuGemmConvolution.cpp b/src/runtime/cpu/operators/CpuGemmConvolution.cpp index 6a78b0cf29..864d7e2d0b 100644 --- a/src/runtime/cpu/operators/CpuGemmConvolution.cpp +++ b/src/runtime/cpu/operators/CpuGemmConvolution.cpp @@ -575,8 +575,7 @@ void CpuGemmConvolution::prepare(ITensorPack &tensors) 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 = { |