diff options
author | Luke Hutton <luke.hutton@arm.com> | 2023-02-22 11:53:48 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-02-28 20:08:57 +0000 |
commit | a4e48ca7b032992ca0110900935c08d7cf860cd3 (patch) | |
tree | a58c8617390225ecc107721d9b5ff87c2bdb01b0 /reference_model/src/ops/op_factory.cc | |
parent | 2226f90d5a6c48a975045bc9e0419113ce764aaf (diff) | |
download | reference_model-a4e48ca7b032992ca0110900935c08d7cf860cd3.tar.gz |
Update rank limits for SLICE, TILE and TRANSPOSE
Updated to align with corresponding changes to the
spec.
In addition, some ERROR_IF tests have been updated to
match the checks specified by the spec, including:
PAD, SLICE, TILE, TRANSPOSE.
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: Ie2c5f48e79a5610eb82739170e25057a63dac1d8
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 |