diff options
Diffstat (limited to 'src/backends/aclCommon')
-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"); } } |