diff options
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: |