diff options
Diffstat (limited to 'reference_model/src/ops/op_factory.cc')
| -rw-r--r-- | reference_model/src/ops/op_factory.cc | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc index a3069dc..d834b74 100644 --- a/reference_model/src/ops/op_factory.cc +++ b/reference_model/src/ops/op_factory.cc @@ -419,6 +419,15 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpPad, BOOL); DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpPad, FP64); break; + case Op_DIM: + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpDim, FP16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpDim, BF16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpDim, FP32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpDim, INT32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpDim, INT8); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpDim, INT16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpDim, BOOL); + break; case Op_RESHAPE: DEF_FACTORY_RESHAPE(OpReshape, FP16); DEF_FACTORY_RESHAPE(OpReshape, BF16); |