From 3112e33d8d0b987e85107390a0350bd5988f5f01 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Wed, 21 Nov 2018 15:44:55 +0000 Subject: COMPMID-1451 Change PriorBox output to NCHw Output of Priorbox should be independent of the input data layout and should always be in NCHW format Change-Id: Ie80cd4e51c78945b158c0db1af1923bdf8d7ea7b --- arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) (limited to 'arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h') diff --git a/arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h b/arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h index ac61b61142..338e9c55b6 100644 --- a/arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h @@ -82,20 +82,13 @@ private: * @param[in] width Input width. * @param[in] height Input height. */ - template void store_coordinates(float *out, const int offset, const float center_x, const float center_y, const float box_width, const float box_height, const int width, const int height); /** Function to calculate prior boxes. * * @param[in] window Input region on which to execute the kernel. */ - template void calculate_prior_boxes(const Window &window); - /** Common signature for all the specialised PriorBox functions - * - * @param[in] window Input region on which to execute the kernel.. - */ - using PriorBoxFunction = void (NEPriorBoxLayerKernel::*)(const Window &window); - PriorBoxFunction _func; + const ITensor *_input1; const ITensor *_input2; ITensor *_output; -- cgit v1.2.1