diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-03-08 10:11:11 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2021-03-08 13:55:35 +0000 |
commit | 023efb30b8ad3dc675c6e2059c82bcfafed4b088 (patch) | |
tree | 74d1913c13f3650822004b55ab60a7616f42d228 | |
parent | b88272e2dcf09803cd275cfeb5ca1ea743a091ff (diff) | |
download | ComputeLibrary-023efb30b8ad3dc675c6e2059c82bcfafed4b088.tar.gz |
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 <georgios.pinitas@arm.com>
Change-Id: I8e5d11a25d5abfc0b2294e2c493deabe47c7dcf4
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5227
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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); |