diff options
Diffstat (limited to 'src/backends/aclCommon')
-rw-r--r-- | src/backends/aclCommon/ArmComputeUtils.hpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp index 2a0707872e..d9efab288f 100644 --- a/src/backends/aclCommon/ArmComputeUtils.hpp +++ b/src/backends/aclCommon/ArmComputeUtils.hpp @@ -255,4 +255,16 @@ inline unsigned int ComputePositiveAxis(const int& axis, const armnn::TensorInfo return static_cast<unsigned int>(positiveAxis); } +inline arm_compute::ReductionOperation ConvertReductionOperationToAcl(const ReduceDescriptor& descriptor) +{ + switch (descriptor.m_ReduceOperation) + { + case ReduceOperation::Sum: return arm_compute::ReductionOperation::SUM; + 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"); + } +} + } // namespace armnn |