aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h
diff options
context:
space:
mode:
authorMichalis Spyrou <michalis.spyrou@arm.com>2018-11-21 15:44:55 +0000
committerMichalis Spyrou <michalis.spyrou@arm.com>2018-11-21 17:32:46 +0000
commit3112e33d8d0b987e85107390a0350bd5988f5f01 (patch)
tree050c8400608caaa216305611ca2de1d8163e462b /arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h
parentedf26ea4921fd0c162abb962fc4863846e73f53f (diff)
downloadComputeLibrary-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/NEPriorBoxLayerKernel.h')
-rw-r--r--arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h9
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;