aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/ops/activation_funcs.h
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/ops/activation_funcs.h')
-rw-r--r--reference_model/src/ops/activation_funcs.h14
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