24.02.1
|
Basic function to execute two dimensional FFT. More...
#include <NEFFT2D.h>
Public Member Functions | |
NEFFT2D (std::shared_ptr< IMemoryManager > memory_manager=nullptr) | |
Default Constructor. More... | |
NEFFT2D (const NEFFT2D &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEFFT2D & | operator= (const NEFFT2D &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEFFT2D (NEFFT2D &&)=delete | |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
NEFFT2D & | operator= (NEFFT2D &&)=delete |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
~NEFFT2D () | |
Default destructor. More... | |
void | configure (const ITensor *input, ITensor *output, const FFT2DInfo &config) |
Initialise the function's source and destinations. More... | |
void | run () override |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
virtual void | prepare () |
Prepare the function for executing. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *input, const ITensorInfo *output, const FFT2DInfo &config) |
Static function to check if given info will lead to a valid configuration of NEFFT2D. More... | |
Basic function to execute two dimensional FFT.
This function calls the following kernels:
NEFFT2D | ( | std::shared_ptr< IMemoryManager > | memory_manager = nullptr | ) |
Default Constructor.
Definition at line 36 of file NEFFT2D.cpp.
Prevent instances of this class from being copied (As this class contains pointers)
Prevent instances of this class from being moved (As this class contains non movable objects)
|
default |
Default destructor.
Initialise the function's source and destinations.
Valid data layouts:
Valid data type configurations:
src | dst |
---|---|
F32 | F32 |
[in] | input | Source tensor. Data types supported: F32. |
[out] | output | Destination tensor. Data types and data layouts supported: Same as input . |
[in] | config | FFT related configuration |
Definition at line 44 of file NEFFT2D.cpp.
References TensorAllocator::allocate(), Tensor::allocator(), ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, ARM_COMPUTE_LOG_PARAMS, FFT1DInfo::axis, FFT2DInfo::axis0, FFT2DInfo::axis1, NEFFT1D::configure(), FFT1DInfo::direction, FFT2DInfo::direction, ITensor::info(), arm_compute::test::validation::input, MemoryGroup::manage(), and NEFFT2D::validate().
Referenced by NEFFTConvolutionLayer::configure().
Prevent instances of this class from being copied (As this class contains pointers)
Prevent instances of this class from being moved (As this class contains non movable objects)
|
overridevirtual |
Run the kernels contained in the function.
For CPU kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 94 of file NEFFT2D.cpp.
References MemoryGroup::acquire(), MemoryGroup::release(), and NEFFT1D::run().
Referenced by NEFFTConvolutionLayer::run().
|
static |
Static function to check if given info will lead to a valid configuration of NEFFT2D.
[in] | input | Source tensor info. Data types supported: F32. |
[in] | output | Destination tensor info. Data types and data layouts supported: Same as input . |
[in] | config | FFT related configuration |
Definition at line 65 of file NEFFT2D.cpp.
References ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES, ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES, ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, ARM_COMPUTE_RETURN_ON_ERROR, FFT1DInfo::axis, FFT2DInfo::axis0, FFT2DInfo::axis1, FFT1DInfo::direction, FFT2DInfo::direction, arm_compute::test::validation::input, ITensorInfo::total_size(), and NEFFT1D::validate().
Referenced by NEFFT2D::configure().