commit 77d0f76b42c6353459d3271ccba9e41728d066cd
author Kevin Cheng <kevin.cheng@arm.com> Tue Nov 24 10:26:32 2020 -0800
committer Kevin Cheng <kevin.cheng@arm.com> Tue Nov 24 14:06:04 2020 -0800
tree 70f95376de80e249a15b15dba6faa27fb4d48f45
parent aee1facbde25caf27cc34e5ec08eb8bba6af8e18

Update reference model/serialization library to 0.21.0 with unit tests added/updated
- update tosa.GATHER
- update tosa.RESIZE
- add tosa.SCATTER

Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I1c3247b831a64e35a85c4044b24c6c29b8e18d25

reference_model/src/ops/op_factory.cc

diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc
index bad0c40..4a06248 100644
--- a/reference_model/src/ops/op_factory.cc
+++ b/reference_model/src/ops/op_factory.cc
@@ -334,14 +334,17 @@
 
         // scatter_gather
         case Op_GATHER:
-        {
-            // output.rank = input.rank - 1 + index.rank
-            int32_t index_rank = outputRank - inputRank + 1;
-            DEF_FACTORY_GATHER(OpGather, AINT8);
-            DEF_FACTORY_GATHER(OpGather, INT16);
-            DEF_FACTORY_GATHER(OpGather, INT32);
-        }
-        break;
+            DEF_FACTORY_ONE_TYPE(OpGather, AINT8);
+            DEF_FACTORY_ONE_TYPE(OpGather, INT16);
+            DEF_FACTORY_ONE_TYPE(OpGather, INT32);
+            DEF_FACTORY_ONE_TYPE(OpGather, FLOAT);
+            break;
+        case Op_SCATTER:
+            DEF_FACTORY_ONE_TYPE(OpScatter, AINT8);
+            DEF_FACTORY_ONE_TYPE(OpScatter, INT16);
+            DEF_FACTORY_ONE_TYPE(OpScatter, INT32);
+            DEF_FACTORY_ONE_TYPE(OpScatter, FLOAT);
+            break;
 
         // image
         case Op_RESIZE:
@@ -349,6 +352,7 @@
             DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, INT8, INT8);
             DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, INT16, INT48);
             DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, INT16, INT16);
+            DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, FLOAT, FLOAT);
             break;
 
         // data_nodes