diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-11-21 15:44:55 +0000 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-11-21 17:32:46 +0000 |
commit | 3112e33d8d0b987e85107390a0350bd5988f5f01 (patch) | |
tree | 050c8400608caaa216305611ca2de1d8163e462b /arm_compute/core/NEON/kernels | |
parent | edf26ea4921fd0c162abb962fc4863846e73f53f (diff) | |
download | ComputeLibrary-3112e33d8d0b987e85107390a0350bd5988f5f01.tar.gz |
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
Diffstat (limited to 'arm_compute/core/NEON/kernels')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h | 9 |
1 files changed, 1 insertions, 8 deletions
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 <DataLayout DL> 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 <DataLayout DL> 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; |