diff options
Diffstat (limited to 'delegate/src/Reduce.hpp')
| -rw-r--r-- | delegate/src/Reduce.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/delegate/src/Reduce.hpp b/delegate/src/Reduce.hpp index 13a11d3e61..bcea16fdb0 100644 --- a/delegate/src/Reduce.hpp +++ b/delegate/src/Reduce.hpp @@ -91,6 +91,10 @@ TfLiteStatus VisitReduceOperator(DelegateData& delegateData, { desc.m_ReduceOperation = armnn::ReduceOperation::Sum; } + else if (reduceOperatorCode == kTfLiteBuiltinReduceProd) + { + desc.m_ReduceOperation = armnn::ReduceOperation::Prod; + } else { TF_LITE_MAYBE_KERNEL_LOG( |