diff options
| author | Won Jeon <won.jeon@arm.com> | 2023-06-10 00:20:04 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-06-29 15:47:40 +0000 |
| commit | 78155c63024c33d31e3b66e8e198fd5f93837df1 (patch) | |
| tree | fcc16affd189eea6d1f4647d8fb4f6ea71a39ff2 /reference_model/src/ops/op_factory.cc | |
| parent | eb74106e1bc52127e5631736e10e8f8b0b7a1d07 (diff) | |
| download | reference_model-78155c63024c33d31e3b66e8e198fd5f93837df1.tar.gz | |
Add support for ERF operator to reference model
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Ib42b867287b83a183a0d0fb1f1eb29974f58fae4
Diffstat (limited to 'reference_model/src/ops/op_factory.cc')
| -rw-r--r-- | reference_model/src/ops/op_factory.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc index 0a78884..a3069dc 100644 --- a/reference_model/src/ops/op_factory.cc +++ b/reference_model/src/ops/op_factory.cc @@ -161,6 +161,12 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTanh, FP32); DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTanh, FP64); break; + case Op_ERF: + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpErf, FP16); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpErf, BF16); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpErf, FP32); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpErf, FP64); + break; // ewise_binary case Op_ADD: |