diff options
Diffstat (limited to 'src/runtime/NEON')
-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); |