diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-08-05 12:34:37 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-09-03 08:41:21 +0000 |
| commit | 4e3e831da1d6d85dffffacf57e9de8fc891b7e58 (patch) | |
| tree | 9a3653729feba788dcfbbdc5255ad379cbbf597d /include | |
| parent | 14bef9f83f7cd58e5038ae7432d75da2d50e7b68 (diff) | |
| download | armnn-4e3e831da1d6d85dffffacf57e9de8fc891b7e58.tar.gz | |
IVGCVSW-6262 Add support for Reduce Prod
* Tflite parser
* Tflite delegate
* Serializer
* Deserializer
* Ref, CpuAcc and GpuAcc workloads
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I601a9ee1680b372c7955d9a628857d08c3cfd377
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/Types.hpp | 3 | ||||
| -rw-r--r-- | include/armnn/TypesUtils.hpp | 1 |
2 files changed, 3 insertions, 1 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 5e00026eba..19e93c3c7e 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -134,7 +134,8 @@ enum class ReduceOperation Sum = 0, Max = 1, Mean = 2, - Min = 3 + Min = 3, + Prod = 4 }; enum class ResizeMethod diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 75cd16341c..20fcbbd784 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -133,6 +133,7 @@ constexpr char const* GetReduceOperationAsCString(ReduceOperation reduce_operati case ReduceOperation::Max: return "Max"; case ReduceOperation::Mean: return "Mean"; case ReduceOperation::Min: return "Min"; + case ReduceOperation::Prod: return "Prod"; default: return "Unknown"; } } |