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
diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp
index 624ce5d..059518a 100644
--- a/src/backends/aclCommon/ArmComputeUtils.hpp
+++ b/src/backends/aclCommon/ArmComputeUtils.hpp
@@ -272,7 +272,8 @@
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");
}
}