From 997aba2dc6a0b5a8c3b424a31ae8370b395cf0d8 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Wed, 28 Jun 2017 11:32:05 +0100 Subject: COMPMID-421: Fixed a problem in Convolution Layer reference values for FP16. All methods in std::numeric_limits return 0. Change-Id: I2289e01853e1b2c38afdec119ef6fc8af8a9752e Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79312 Reviewed-by: Anthony Barbier Tested-by: Kaizen Reviewed-by: Moritz Pflanzer --- src/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.cpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.cpp') diff --git a/src/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.cpp b/src/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.cpp index dcfbb13081..1db025723c 100644 --- a/src/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.cpp +++ b/src/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.cpp @@ -639,6 +639,7 @@ template void matrix_matrix_multiply_f16(const ITensor *input0, const ITensor *input1, ITensor *output, const Window &window, float alpha) { #ifdef ARM_COMPUTE_ENABLE_FP16 + const size_t in_b_stride = input1->info()->strides_in_bytes()[1] / data_size_from_type(input1->info()->data_type()); const size_t out_stride = output->info()->strides_in_bytes()[1] / data_size_from_type(output->info()->data_type()); -- cgit v1.2.1