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