diff options
Diffstat (limited to 'reference_model/src/ops/reduction.h')
-rw-r--r-- | reference_model/src/ops/reduction.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/reference_model/src/ops/reduction.h b/reference_model/src/ops/reduction.h index f4e29b9..f3407f4 100644 --- a/reference_model/src/ops/reduction.h +++ b/reference_model/src/ops/reduction.h @@ -104,6 +104,16 @@ public: virtual int eval(); }; +template <int Rank, DType Dtype> +class OpReduceSumInt : public ReduceNode<Rank, Dtype> +{ +public: + OpReduceSumInt(SubgraphTraverser* sgt_, TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_) + : ReduceNode<Rank, Dtype>(sgt_, Op_REDUCE_SUM, attribute_, id_) + {} + virtual int eval(); +}; + }; // namespace TosaReference #endif |