diff options
Diffstat (limited to 'reference_model/src/ops/activation_funcs.h')
-rw-r--r-- | reference_model/src/ops/activation_funcs.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/reference_model/src/ops/activation_funcs.h b/reference_model/src/ops/activation_funcs.h index 2372fcb..a7e1275 100644 --- a/reference_model/src/ops/activation_funcs.h +++ b/reference_model/src/ops/activation_funcs.h @@ -77,6 +77,20 @@ public: virtual int register_fcn(); }; +template <int Rank, TOSA_REF_TYPE Dtype> +class OpErf : public UnaryNode<Rank, Dtype> +{ +public: + OpErf(SubgraphTraverser* sgt_, TosaAttributeBase* attribute_, uint64_t id_) + : UnaryNode<Rank, Dtype>(sgt_, Op_ERF, id_) + { + register_fcn(); + } + using InEigenType = typename GetEigenType<Dtype>::type; + using OutEigenType = typename GetEigenType<Dtype>::type; + virtual int register_fcn(); +}; + }; // namespace TosaReference #endif |