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 /src/backends/aclCommon/ArmComputeUtils.hpp | |
| 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 'src/backends/aclCommon/ArmComputeUtils.hpp')
| -rw-r--r-- | src/backends/aclCommon/ArmComputeUtils.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp index 624ce5df7a..059518acd6 100644 --- a/src/backends/aclCommon/ArmComputeUtils.hpp +++ b/src/backends/aclCommon/ArmComputeUtils.hpp @@ -272,7 +272,8 @@ inline arm_compute::ReductionOperation ConvertReductionOperationToAcl(const Redu case ReduceOperation::Mean: return arm_compute::ReductionOperation::MEAN_SUM; case ReduceOperation::Max: return arm_compute::ReductionOperation::MAX; case ReduceOperation::Min: return arm_compute::ReductionOperation::MIN; - default: throw InvalidArgumentException("Unsupported Reduction operation"); + case ReduceOperation::Prod: return arm_compute::ReductionOperation::PROD; + default: throw InvalidArgumentException("Unsupported Reduction operation"); } } |