diff options
author | evacha01 <evan.chandler@arm.com> | 2024-03-06 17:33:44 +0000 |
---|---|---|
committer | evacha01 <evan.chandler@arm.com> | 2024-03-07 12:52:13 +0000 |
commit | 1adff8301ac7e4933382549ad22b3b47cdb01eda (patch) | |
tree | 1cbe95bb38a9f877e5bbc89294064239e3e4b8b4 /reference_model/src/ops/data_nodes.cc | |
parent | 9c96eefbaca6c85be79529bce7ff04fd7dfe3a0d (diff) | |
download | reference_model-1adff8301ac7e4933382549ad22b3b47cdb01eda.tar.gz |
Add INT48 and INT4 datatype support to IDENTITY
Added support to ref model and testing
Signed-off-by: evacha01 <evan.chandler@arm.com>
Change-Id: Iece53d07c8986332fdd8f1ce5ed6265349df1b6a
Diffstat (limited to 'reference_model/src/ops/data_nodes.cc')
-rw-r--r-- | reference_model/src/ops/data_nodes.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/reference_model/src/ops/data_nodes.cc b/reference_model/src/ops/data_nodes.cc index 64001a9..8b480ee 100644 --- a/reference_model/src/ops/data_nodes.cc +++ b/reference_model/src/ops/data_nodes.cc @@ -100,9 +100,11 @@ int OpIdentity<Rank, Dtype>::eval() DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, FP16); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, BF16); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, FP32); +DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT4); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT8); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT16); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT32); +DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT48); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, BOOL); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, FP64); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, FP8E4M3); |