From e46a7beb6b3d89b7cc6a96faeab19ee4478d4e2e Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 18 Feb 2019 15:16:14 +0000 Subject: COMPMID-2005: Tensors have different quantization information. NEGEMMInterleavedKernel quantization information were unset when using the respective convolution functions. Change-Id: I1fd2af33aaadcfe3eda8bf20a0e56afa9b77c9bb Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/722 Reviewed-by: Isabella Gottardi Tested-by: Arm Jenkins --- src/runtime/NEON/functions/NEGEMM.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/runtime/NEON/functions/NEGEMM.cpp') diff --git a/src/runtime/NEON/functions/NEGEMM.cpp b/src/runtime/NEON/functions/NEGEMM.cpp index 72a3e80330..99cddc2920 100644 --- a/src/runtime/NEON/functions/NEGEMM.cpp +++ b/src/runtime/NEON/functions/NEGEMM.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -91,8 +91,8 @@ void NEGEMM::configure(const ITensor *a, const ITensor *b, const ITensor *c, ITe shape_tmp_b.set(0, b->info()->dimension(1) * transpose_w); shape_tmp_b.set(1, std::ceil(b->info()->dimension(0) / static_cast(transpose_w))); - TensorInfo info_a(shape_tmp_a, 1, a->info()->data_type()); - TensorInfo info_b(shape_tmp_b, 1, b->info()->data_type()); + TensorInfo info_a = a->info()->clone()->set_tensor_shape(shape_tmp_a); + TensorInfo info_b = b->info()->clone()->set_tensor_shape(shape_tmp_b); _tmp_a.allocator()->init(info_a); _tmp_b.allocator()->init(info_b); -- cgit v1.2.1