diff options
Diffstat (limited to 'reference_model/src/ops/op_factory.cc')
| -rw-r--r-- | reference_model/src/ops/op_factory.cc | 42 |
1 files changed, 21 insertions, 21 deletions
diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc index 8d84135..1db3974 100644 --- a/reference_model/src/ops/op_factory.cc +++ b/reference_model/src/ops/op_factory.cc @@ -394,31 +394,31 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReverse, BOOL); break; case Op_SLICE: - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, FP16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, BF16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, FP32); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, INT8); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, INT16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, INT32); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, BOOL); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpSlice, FP16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpSlice, BF16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpSlice, FP32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpSlice, INT8); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpSlice, INT16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpSlice, INT32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpSlice, BOOL); break; case Op_TILE: - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, FP16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, BF16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, FP32); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, INT8); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, INT16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, INT32); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, BOOL); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTile, FP16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTile, BF16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTile, FP32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTile, INT8); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTile, INT16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTile, INT32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTile, BOOL); break; case Op_TRANSPOSE: - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, BOOL); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, FP16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, BF16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, FP32); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, INT8); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, INT16); - DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, INT32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTranspose, BOOL); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTranspose, FP16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTranspose, BF16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTranspose, FP32); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTranspose, INT8); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTranspose, INT16); + DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpTranspose, INT32); break; // scatter_gather |