diff options
| author | Jerry Ge <jerry.ge@arm.com> | 2024-02-20 11:21:19 -0800 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2024-03-05 18:47:09 +0000 |
| commit | 51bd4f5166c50a89017307b55dee0f5bda096b7b (patch) | |
| tree | 84aa5e9bd1dc02856ae10a18f0a923e3c8efbf55 /reference_model/src/ops/op_factory.cc | |
| parent | 1408795800719139e26bafcece88bfc07582576d (diff) | |
| download | reference_model-51bd4f5166c50a89017307b55dee0f5bda096b7b.tar.gz | |
Add Tosa Sin/Cos operators
- Add Tosa Sin/Cos operators to reference_model
- Add conformances tests
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I3f597ddf5dac2c64d6dd6aa15781b40b8468eaa6
Diffstat (limited to 'reference_model/src/ops/op_factory.cc')
| -rw-r--r-- | reference_model/src/ops/op_factory.cc | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc index 6d66c07..1891ff4 100644 --- a/reference_model/src/ops/op_factory.cc +++ b/reference_model/src/ops/op_factory.cc @@ -299,6 +299,12 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, case Op_CLZ: DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpClz, INT32); break; + case Op_COS: + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpCos, FP16); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpCos, BF16); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpCos, FP32); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpCos, FP64); + break; case Op_EXP: DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpExp, FP16); DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpExp, BF16); @@ -341,6 +347,12 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpRsqrt, FP32); DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpRsqrt, FP64); break; + case Op_SIN: + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSin, FP16); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSin, BF16); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSin, FP32); + DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSin, FP64); + break; // ewise_ternary case Op_SELECT: |