24 #ifndef __ARM_COMPUTE_NEMAGNITUDEPHASEKERNEL_H__ 25 #define __ARM_COMPUTE_NEMAGNITUDEPHASEKERNEL_H__ 35 template <MagnitudeType mag_type, PhaseType phase_type>
39 const char *
name()
const override 41 return "NEMagnitudePhaseKernel";
75 void magnitude(
const Window &window);
80 void phase(
const Window &window);
85 void magnitude_phase(
const Window &window);
94 MagnitudePhaseFunctionPtr _func;
101 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC 103 template <MagnitudeType mag_type, PhaseType phase_type>
107 const char *
name()
const override 109 return "NEMagnitudePhaseFP16Kernel";
143 void magnitude(
const Window &window);
148 void phase(
const Window &window);
161 MagnitudePhaseFunctionPtr _func;
169 template <MagnitudeType mag_type, PhaseType phase_type>
NEMagnitudePhaseKernel()
Default constructor.
const char * name() const override
Name of the kernel.
Common interface for all kernels implemented in C++.
NEMagnitudePhaseKernel & operator=(const NEMagnitudePhaseKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Template interface for the kernel to compute magnitude and phase.
NEMagnitudePhaseKernel< mag_type, phase_type > NEMagnitudePhaseFP16Kernel
Template interface for the kernel to compute magnitude and phase.
Interface for NEON tensor.
src info() -> set_format(Format::S16)
This file contains all available output stages for GEMMLowp on OpenCL.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
~NEMagnitudePhaseKernel()=default
Destructor.
Information about executing thread and CPU.
__kernel void magnitude_phase(__global uchar *gx_ptr, uint gx_stride_x, uint gx_step_x, uint gx_stride_y, uint gx_step_y, uint gx_offset_first_element_in_bytes, __global uchar *gy_ptr, uint gy_stride_x, uint gy_step_x, uint gy_stride_y, uint gy_step_y, uint gy_offset_first_element_in_bytes, __global uchar *magnitude_ptr, uint magnitude_stride_x, uint magnitude_step_x, uint magnitude_stride_y, uint magnitude_step_y, uint magnitude_offset_first_element_in_bytes, __global uchar *phase_ptr, uint phase_stride_x, uint phase_step_x, uint phase_stride_y, uint phase_step_y, uint phase_offset_first_element_in_bytes)
Calculate the magnitude and phase of given the gradients of an image.
void configure(const ITensor *gx, const ITensor *gy, ITensor *magnitude, ITensor *phase)
Initialise the kernel's input, output.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.