diff options
| author | James Ward <james.ward@arm.com> | 2022-08-12 20:48:56 +0100 |
|---|---|---|
| committer | James Ward <james.ward@arm.com> | 2022-10-11 11:56:02 +0100 |
| commit | 8b39043c70332e1e4c95ee6a9616aec40dd3baf1 (patch) | |
| tree | fea519246b698eb944b9d58537fc90bc30481d11 /reference_model/src/ops/scatter_gather.cc | |
| parent | ba5fad356a926d5e1c6e0fe6b546a310230cc5a8 (diff) | |
| download | reference_model-8b39043c70332e1e4c95ee6a9616aec40dd3baf1.tar.gz | |
Reference model changes for fp16 support
Change-Id: I72f21fcfa153046274969d327313e3349981dbe6
Signed-off-by: James Ward <james.ward@arm.com>
Diffstat (limited to 'reference_model/src/ops/scatter_gather.cc')
| -rw-r--r-- | reference_model/src/ops/scatter_gather.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/reference_model/src/ops/scatter_gather.cc b/reference_model/src/ops/scatter_gather.cc index faf7db9..25174bd 100644 --- a/reference_model/src/ops/scatter_gather.cc +++ b/reference_model/src/ops/scatter_gather.cc @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2021, ARM Limited. +// Copyright (c) 2020-2022, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -226,9 +226,11 @@ int OpScatter<Dtype>::eval() DEF_INSTANTIATE_ONE_TYPE(OpGather, INT8); DEF_INSTANTIATE_ONE_TYPE(OpGather, INT16); DEF_INSTANTIATE_ONE_TYPE(OpGather, INT32); +DEF_INSTANTIATE_ONE_TYPE(OpGather, FP16); DEF_INSTANTIATE_ONE_TYPE(OpGather, FLOAT); DEF_INSTANTIATE_ONE_TYPE(OpScatter, INT8); DEF_INSTANTIATE_ONE_TYPE(OpScatter, INT16); DEF_INSTANTIATE_ONE_TYPE(OpScatter, INT32); +DEF_INSTANTIATE_ONE_TYPE(OpScatter, FP16); DEF_INSTANTIATE_ONE_TYPE(OpScatter, FLOAT); |