diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEFFT2D.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEFFT2D.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/runtime/NEON/functions/NEFFT2D.cpp b/src/runtime/NEON/functions/NEFFT2D.cpp index 9210ecfa2e..2fea017781 100644 --- a/src/runtime/NEON/functions/NEFFT2D.cpp +++ b/src/runtime/NEON/functions/NEFFT2D.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019 ARM Limited. + * Copyright (c) 2019-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -41,14 +41,14 @@ void NEFFT2D::configure(const ITensor *input, ITensor *output, const FFT2DInfo & // Setup first pass FFT1DInfo first_pass_config; - first_pass_config.axis = config.axes.first; + first_pass_config.axis = config.axis0; first_pass_config.direction = config.direction; _memory_group.manage(&_first_pass_tensor); _first_pass_func.configure(input, &_first_pass_tensor, first_pass_config); // Setup second pass FFT1DInfo second_pass_config; - second_pass_config.axis = config.axes.second; + second_pass_config.axis = config.axis1; second_pass_config.direction = config.direction; _second_pass_func.configure(&_first_pass_tensor, output, second_pass_config); _first_pass_tensor.allocator()->allocate(); @@ -63,13 +63,13 @@ Status NEFFT2D::validate(const ITensorInfo *input, const ITensorInfo *output, co // Validate first pass FFT1DInfo first_pass_config; - first_pass_config.axis = config.axes.first; + first_pass_config.axis = config.axis0; first_pass_config.direction = config.direction; ARM_COMPUTE_RETURN_ON_ERROR(NEFFT1D::validate(input, &first_pass_tensor, first_pass_config)); // Validate second pass FFT1DInfo second_pass_config; - second_pass_config.axis = config.axes.second; + second_pass_config.axis = config.axis1; second_pass_config.direction = config.direction; ARM_COMPUTE_RETURN_ON_ERROR(NEFFT1D::validate(&first_pass_tensor, output, second_pass_config)); |