diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEGenerateProposalsLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEGenerateProposalsLayer.h | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEGenerateProposalsLayer.h b/arm_compute/runtime/NEON/functions/NEGenerateProposalsLayer.h index 0e6601e4f9..c6d3628e37 100644 --- a/arm_compute/runtime/NEON/functions/NEGenerateProposalsLayer.h +++ b/arm_compute/runtime/NEON/functions/NEGenerateProposalsLayer.h @@ -25,9 +25,8 @@ #define __ARM_COMPUTE_NEGENERATEPROPOSALSLAYER_H__ #include "arm_compute/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.h" #include "arm_compute/core/NEON/kernels/NEBoundingBoxTransformKernel.h" -#include "arm_compute/core/NEON/kernels/NECopyKernel.h" #include "arm_compute/core/NEON/kernels/NEGenerateProposalsLayerKernel.h" -#include "arm_compute/core/NEON/kernels/NEMemsetKernel.h" +#include "arm_compute/core/NEON/kernels/NEPadLayerKernel.h" #include "arm_compute/core/NEON/kernels/NEPermuteKernel.h" #include "arm_compute/core/NEON/kernels/NEReshapeLayerKernel.h" #include "arm_compute/core/NEON/kernels/NEStridedSliceKernel.h" @@ -49,8 +48,7 @@ class ITensor; * -# @ref NEReshapeLayer x 2 * -# @ref NEStridedSlice x 3 * -# @ref NEBoundingBoxTransform - * -# @ref NECopyKernel - * -# @ref NEMemsetKernel + * -# @ref NEPadLayerKernel * And the following CPP kernels: * -# @ref CPPBoxWithNonMaximaSuppressionLimit */ @@ -117,8 +115,7 @@ private: NEReshapeLayerKernel _flatten_scores_kernel; NEComputeAllAnchorsKernel _compute_anchors_kernel; NEBoundingBoxTransformKernel _bounding_box_kernel; - NEMemsetKernel _memset_kernel; - NECopyKernel _padded_copy_kernel; + NEPadLayerKernel _pad_kernel; // CPP kernels CPPBoxWithNonMaximaSuppressionLimitKernel _cpp_nms_kernel; |