diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h | 3 |
1 files changed, 2 insertions, 1 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. |