• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2016-2020 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #ifndef ARM_COMPUTE_CLGAUSSIAN5X5KERNEL_H
25 #define ARM_COMPUTE_CLGAUSSIAN5X5KERNEL_H
26 
27 #include "src/core/CL/kernels/CLConvolutionKernel.h"
28 
29 namespace arm_compute
30 {
31 class ICLTensor;
32 
33 /** Interface for the kernel to run the horizontal pass of 5x5 Gaussian filter on a tensor. */
34 class CLGaussian5x5HorKernel : public CLSeparableConvolution5x5HorKernel
35 {
36 public:
37     /** Initialise the kernel's source, destination and border.
38      *
39      * @param[in]  input            Source tensor. Data types supported: U8.
40      * @param[out] output           Destination tensor. Data types supported: S16.
41      * @param[in]  border_undefined True if the border mode is undefined. False if it's replicate or constant.
42      */
43     void configure(const ICLTensor *input, ICLTensor *output, bool border_undefined);
44     /** Initialise the kernel's source, destination and border.
45      *
46      * @param[in]  compile_context  The compile context to be used.
47      * @param[in]  input            Source tensor. Data types supported: U8.
48      * @param[out] output           Destination tensor. Data types supported: S16.
49      * @param[in]  border_undefined True if the border mode is undefined. False if it's replicate or constant.
50      */
51     void configure(const CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, bool border_undefined);
52 
53 private:
54     //Make the configure method of the parent class private
55     using CLSeparableConvolution5x5HorKernel::configure;
56 };
57 
58 /** Interface for the kernel to run the vertical pass of 5x5 Gaussian filter on a tensor. */
59 class CLGaussian5x5VertKernel : public CLSeparableConvolution5x5VertKernel
60 {
61 public:
62     /** Initialise the kernel's source, destination and border.
63      *
64      * @param[in]  input            Input tensor(output of horizontal pass). Data types supported: S16.
65      * @param[out] output           Destination tensor. Data types supported: U8.
66      * @param[in]  border_undefined True if the border mode is undefined. False if it's replicate or constant.
67      */
68     void configure(const ICLTensor *input, ICLTensor *output, bool border_undefined);
69     /** Initialise the kernel's source, destination and border.
70      *
71      * @param[in]  compile_context  The compile context to be used.
72      * @param[in]  input            Input tensor(output of horizontal pass). Data types supported: S16.
73      * @param[out] output           Destination tensor. Data types supported: U8.
74      * @param[in]  border_undefined True if the border mode is undefined. False if it's replicate or constant.
75      */
76     void configure(const CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, bool border_undefined);
77 
78 private:
79     //Make the configure method of the parent class private
80     using CLSeparableConvolution5x5VertKernel::configure;
81 };
82 } // namespace arm_compute
83 #endif /*ARM_COMPUTE_CLGAUSSIAN5X5KERNEL_H */
84