commit 61a8313f5a0cbcfc7c8ee8a44f05a5ca9b1015b9
author Jerry Ge <jerry.ge@arm.com> Tue Feb 20 19:03:06 2024 +0000
committer Jerry Ge <jerry.ge@arm.com> Tue Feb 20 19:05:02 2024 +0000
tree d15f44d2d7c17288dcbde6ad53e360b724269830
parent 7eb878c434a8eeaf2f6e18325bdcdd535c86c080

Add Tosa Sin/Cos to schema

- Add Tosa Sin/Cos to the serialization_lib schema
- Generated related header files with the new schema

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: If424ff2b64538e8ffd5342b6ebb9a022537d5c4c

include/tosa_generated.h

diff --git a/include/tosa_generated.h b/include/tosa_generated.h
index b5a8bd5..d21a17a 100644
--- a/include/tosa_generated.h
+++ b/include/tosa_generated.h
@@ -284,11 +284,13 @@
   Op_SUB_SHAPE = 76,
   Op_MUL_SHAPE = 77,
   Op_DIV_SHAPE = 78,
+  Op_COS = 79,
+  Op_SIN = 80,
   Op_MIN = Op_UNKNOWN,
-  Op_MAX = Op_DIV_SHAPE
+  Op_MAX = Op_SIN
 };
 
-inline const Op (&EnumValuesOp())[79] {
+inline const Op (&EnumValuesOp())[81] {
   static const Op values[] = {
     Op_UNKNOWN,
     Op_ARGMAX,
@@ -368,13 +370,15 @@
     Op_ADD_SHAPE,
     Op_SUB_SHAPE,
     Op_MUL_SHAPE,
-    Op_DIV_SHAPE
+    Op_DIV_SHAPE,
+    Op_COS,
+    Op_SIN
   };
   return values;
 }
 
 inline const char * const *EnumNamesOp() {
-  static const char * const names[80] = {
+  static const char * const names[82] = {
     "UNKNOWN",
     "ARGMAX",
     "AVG_POOL2D",
@@ -454,13 +458,15 @@
     "SUB_SHAPE",
     "MUL_SHAPE",
     "DIV_SHAPE",
+    "COS",
+    "SIN",
     nullptr
   };
   return names;
 }
 
 inline const char *EnumNameOp(Op e) {
-  if (::flatbuffers::IsOutRange(e, Op_UNKNOWN, Op_DIV_SHAPE)) return "";
+  if (::flatbuffers::IsOutRange(e, Op_UNKNOWN, Op_SIN)) return "";
   const size_t index = static_cast<size_t>(e);
   return EnumNamesOp()[index];
 }

python/tosa/Op.py

diff --git a/python/tosa/Op.py b/python/tosa/Op.py
index c34f109..35b2b80 100644
--- a/python/tosa/Op.py
+++ b/python/tosa/Op.py
@@ -82,3 +82,5 @@
     SUB_SHAPE = 76
     MUL_SHAPE = 77
     DIV_SHAPE = 78
+    COS = 79
+    SIN = 80

schema/tosa.fbs

diff --git a/schema/tosa.fbs b/schema/tosa.fbs
index 1c2a85e..e37c89a 100644
--- a/schema/tosa.fbs
+++ b/schema/tosa.fbs
@@ -127,6 +127,8 @@
   SUB_SHAPE,
   MUL_SHAPE,
   DIV_SHAPE,
+  COS,
+  SIN,
 }
 
 union Attribute {