diff options
| author | 장지섭/On-Device Lab(SR)/Engineer/삼성전자 <jiseob.jang@samsung.com> | 2019-09-17 17:23:58 +0900 |
|---|---|---|
| committer | 이춘석/On-Device Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com> | 2019-09-17 17:23:58 +0900 |
| commit | 4d4f85e1561eaf7dfd9f4126e24b85af24168d55 (patch) | |
| tree | 2374e26b92116afca22c8976ff5dd0e1c1facf5e /runtimes/libs/ARMComputeEx/arm_compute/runtime/NEON/NEFunctionsEx.h | |
| parent | aa5f9788dc53e35360cdaea6e219d20d1cf23c5f (diff) | |
| download | nnfw-4d4f85e1561eaf7dfd9f4126e24b85af24168d55.tar.gz nnfw-4d4f85e1561eaf7dfd9f4126e24b85af24168d55.tar.bz2 nnfw-4d4f85e1561eaf7dfd9f4126e24b85af24168d55.zip | |
Make to support ArgMax op for acl neon (#7515)
This commit makes to support ArgMax op for acl neon except int32 type.
- Introduce NEArgMinMax layer
- Apply NEArgMinMax layer for neurun
Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>
Diffstat (limited to 'runtimes/libs/ARMComputeEx/arm_compute/runtime/NEON/NEFunctionsEx.h')
| -rw-r--r-- | runtimes/libs/ARMComputeEx/arm_compute/runtime/NEON/NEFunctionsEx.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/runtimes/libs/ARMComputeEx/arm_compute/runtime/NEON/NEFunctionsEx.h b/runtimes/libs/ARMComputeEx/arm_compute/runtime/NEON/NEFunctionsEx.h index 6eb083043..fb5323d22 100644 --- a/runtimes/libs/ARMComputeEx/arm_compute/runtime/NEON/NEFunctionsEx.h +++ b/runtimes/libs/ARMComputeEx/arm_compute/runtime/NEON/NEFunctionsEx.h @@ -16,6 +16,7 @@ #ifndef __ARM_COMPUTE_NEFUNCTIONSEX_H__ #define __ARM_COMPUTE_NEFUNCTIONSEX_H__ +#include <arm_compute/runtime/NEON/functions/NEArgMinMax.h> #include <arm_compute/runtime/NEON/functions/NEBinaryLogicalOperation.h> #include <arm_compute/runtime/NEON/functions/NEFullyConnectedReshapingLayer.h> #include <arm_compute/runtime/NEON/functions/NEPReLU.h> |