From 023efb30b8ad3dc675c6e2059c82bcfafed4b088 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 8 Mar 2021 10:11:11 +0000 Subject: Make auxiliary tensor within NEGEMMConvolutionLayer resizable Internal tensor that is used to handle is now set to resizable explicitly. This resolves functional issues in cases the output tensor is marked as non resizable (a common case for imported memory use-cases). Resolves: COMPMID-4294 Signed-off-by: Georgios Pinitas Change-Id: I8e5d11a25d5abfc0b2294e2c493deabe47c7dcf4 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5227 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio --- src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/runtime/NEON') diff --git a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp index e623d9dc74..e2db6b48cd 100644 --- a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp @@ -373,7 +373,7 @@ void NEGEMMConvolutionLayer::configure(const ITensor *input, const ITensor *weig else { TensorInfo out_info{ *output->info() }; - out_info.set_data_type(output_data_type).set_data_layout(input->info()->data_layout()); + out_info.set_data_type(output_data_type).set_data_layout(input->info()->data_layout()).set_is_resizable(true); _gemm_output.allocator()->init(out_info); _gemm_output_3d.allocator()->init(out_info); _memory_group.manage(&_gemm_output); -- cgit v1.2.1