diff options
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/NEON/kernels/NEGaussianPyramidKernel.h | 17 | ||||
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEGaussianPyramid.h | 3 |
2 files changed, 9 insertions, 11 deletions
diff --git a/arm_compute/core/NEON/kernels/NEGaussianPyramidKernel.h b/arm_compute/core/NEON/kernels/NEGaussianPyramidKernel.h index 31779b520c..d28501107d 100644 --- a/arm_compute/core/NEON/kernels/NEGaussianPyramidKernel.h +++ b/arm_compute/core/NEON/kernels/NEGaussianPyramidKernel.h @@ -49,19 +49,17 @@ public: /** Initialise the kernel's source, destination and border mode. * - * @param[in] input Source tensor. Data type supported: U8. - * @param[out] output Destination tensor. Data type supported: S16. - * @param[in] border_undefined True if the border mode is undefined. False if it's replicate or constant. + * @param[in] input Source tensor. Data type supported: U8. + * @param[out] output Destination tensor. Data type supported: S16. */ - void configure(const ITensor *input, ITensor *output, bool border_undefined); + void configure(const ITensor *input, ITensor *output); // Inherited methods overridden: void run(const Window &window, const ThreadInfo &info) override; BorderSize border_size() const override; private: - BorderSize _border_size; - int _l2_load_offset; + int _l2_load_offset; }; /** NEON kernel to perform a GaussianPyramid (vertical pass) */ @@ -83,11 +81,10 @@ public: /** Initialise the kernel's source, destination and border mode. * - * @param[in] input Source tensor. Data type supported: S16. - * @param[out] output Destination tensor. Data type supported: U8. - * @param[in] border_undefined True if the border mode is undefined. False if it's replicate or constant. + * @param[in] input Source tensor. Data type supported: S16. + * @param[out] output Destination tensor. Data type supported: U8. */ - void configure(const ITensor *input, ITensor *output, bool border_undefined); + void configure(const ITensor *input, ITensor *output); // Inherited methods overridden: void run(const Window &window, const ThreadInfo &info) override; diff --git a/arm_compute/runtime/NEON/functions/NEGaussianPyramid.h b/arm_compute/runtime/NEON/functions/NEGaussianPyramid.h index b4ed56a0c3..dbe0ecdf66 100644 --- a/arm_compute/runtime/NEON/functions/NEGaussianPyramid.h +++ b/arm_compute/runtime/NEON/functions/NEGaussianPyramid.h @@ -91,7 +91,8 @@ public: void run() override; private: - std::unique_ptr<NEFillBorderKernel[]> _border_handler; + std::unique_ptr<NEFillBorderKernel[]> _horizontal_border_handler; + std::unique_ptr<NEFillBorderKernel[]> _vertical_border_handler; std::unique_ptr<NEGaussianPyramidHorKernel[]> _horizontal_reduction; std::unique_ptr<NEGaussianPyramidVertKernel[]> _vertical_reduction; }; |