24.02
|
Go to the documentation of this file.
37 : _memory_group(memory_manager),
38 _first_pass_func(memory_manager),
39 _second_pass_func(memory_manager),
54 _memory_group.
manage(&_first_pass_tensor);
55 _first_pass_func.
configure(
input, &_first_pass_tensor, first_pass_config);
61 _second_pass_func.
configure(&_first_pass_tensor, output, second_pass_config);
70 TensorInfo first_pass_tensor(
input->clone()->set_is_resizable(
true).reset_padding().set_num_channels(2));
85 if ((output !=
nullptr) && (output->
total_size() != 0))
98 _first_pass_func.
run();
99 _second_pass_func.
run();
void manage(IMemoryManageable *obj) override
Sets a object to be managed by the given memory group.
void configure(const ITensor *input, ITensor *output, const FFT1DInfo &config)
Initialise the function's source and destinations.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(...)
NEFFT2D(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default Constructor.
Interface for CPU tensor.
void configure(const ITensor *input, ITensor *output, const FFT2DInfo &config)
Initialise the function's source and destinations.
void run() override
Run the kernels contained in the function.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(...)
void acquire() override
Acquires backing memory for the whole group.
#define ARM_COMPUTE_RETURN_ON_ERROR(status)
Checks if a status contains an error and returns it.
void release() override
Releases backing memory of the whole group.
#define ARM_COMPUTE_ERROR_ON_NULLPTR(...)
FFTDirection direction
Direction of the FFT.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
TensorAllocator * allocator()
Return a pointer to the tensor's allocator.
#define ARM_COMPUTE_ERROR_THROW_ON(status)
unsigned int axis0
Axis to run first pass on.
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.
unsigned int axis
Axis to run the FFT on.
Descriptor used by the FFT2D function.
Descriptor used by the FFT1D function.
Store the tensor's metadata.
FFTDirection direction
Direction of the FFT.
~NEFFT2D()
Default destructor.
void allocate() override
Allocate size specified by TensorInfo of CPU memory.
Copyright (c) 2017-2024 Arm Limited.
#define ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(...)
Store the tensor's metadata.
unsigned int axis1
Axis to run second pass on.
void run() override
Run the kernels contained in the function.
#define ARM_COMPUTE_LOG_PARAMS(...)
virtual size_t total_size() const =0
Returns the total size of the tensor in bytes.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const FFT1DInfo &config)
Static function to check if given info will lead to a valid configuration of NEFFT1D.