diff options
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h | 3 | ||||
-rw-r--r-- | src/runtime/NEON/functions/NEArgMinMaxLayer.cpp | 1 |
2 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h b/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h index b0e2d783b3..d2ddff2665 100644 --- a/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h +++ b/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h @@ -24,6 +24,8 @@ #ifndef __ARM_COMPUTE_NEARGMINMAXLAYER_H__ #define __ARM_COMPUTE_NEARGMINMAXLAYER_H__ +#include "arm_compute/runtime/NEON/functions/NEReductionOperation.h" + #include "arm_compute/core/Types.h" #include "arm_compute/runtime/MemoryGroup.h" #include "arm_compute/runtime/NEON/INESimpleFunction.h" @@ -31,7 +33,6 @@ namespace arm_compute { class ITensor; -class NEReductionOperation; /** Function to calculate the index of the minimum or maximum values in a * tensor based on an axis. diff --git a/src/runtime/NEON/functions/NEArgMinMaxLayer.cpp b/src/runtime/NEON/functions/NEArgMinMaxLayer.cpp index ab2d6f0c1f..a23061e296 100644 --- a/src/runtime/NEON/functions/NEArgMinMaxLayer.cpp +++ b/src/runtime/NEON/functions/NEArgMinMaxLayer.cpp @@ -23,7 +23,6 @@ */ #include "arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h" -#include "arm_compute/runtime/NEON/functions/NEReductionOperation.h" #include "arm_compute/core/Error.h" #include "arm_compute/core/ITensor.h" |