diff options
Diffstat (limited to 'arm_compute/runtime/FunctionDescriptors.h')
| -rw-r--r-- | arm_compute/runtime/FunctionDescriptors.h | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/arm_compute/runtime/FunctionDescriptors.h b/arm_compute/runtime/FunctionDescriptors.h index 7ff25019e6..f9b16e4218 100644 --- a/arm_compute/runtime/FunctionDescriptors.h +++ b/arm_compute/runtime/FunctionDescriptors.h @@ -24,12 +24,29 @@ #ifndef __ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H__ #define __ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H__ +#include <utility> + namespace arm_compute { -/** Descriptor used by the FFT1d function */ +/** FFT direction to use */ +enum class FFTDirection +{ + Forward, + Inverse +}; + +/** Descriptor used by the FFT1D function */ struct FFT1DInfo { - unsigned int axis{ 0 }; /**< Axis to run the FFT on. */ + unsigned int axis{ 0 }; /**< Axis to run the FFT on. */ + FFTDirection direction{ FFTDirection::Forward }; /**< Direction of the FFT. */ +}; + +/** Descriptor used by the FFT2D function */ +struct FFT2DInfo +{ + std::pair<unsigned int, unsigned int> axes{ 0, 1 }; /**< Axes to run on. If same, multiple transforms are performed on single axis*/ + FFTDirection direction{ FFTDirection::Forward }; /**< Direction of the FFT. */ }; } // namespace arm_compute #endif /* __ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H__ */ |