Add FFT2d and RFFT2d serialization support
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: I79b3684ae1cf24e71ec7098ed7db5be5bc2f6fb2
diff --git a/schema/tosa.fbs b/schema/tosa.fbs
index 093c235..ff8697b 100644
--- a/schema/tosa.fbs
+++ b/schema/tosa.fbs
@@ -20,6 +20,9 @@
// File extension of any written files.
file_extension "tosa";
+// NOTE: New values added to the schema should be placed
+// at the end of the list in order to keep schema stable.
+
enum DType:uint32 {
UNKNOWN = 0,
BOOL,
@@ -43,8 +46,6 @@
enum Op:uint32 {
UNKNOWN = 0,
-
- // Tensor Operator
ARGMAX,
AVG_POOL2D,
CONV2D,
@@ -54,14 +55,10 @@
MATMUL,
MAX_POOL2D,
TRANSPOSE_CONV2D,
-
- // Activation
CLAMP,
RESERVED,
SIGMOID,
TANH,
-
- // Elementwise-Binary
ADD,
ARITHMETIC_RIGHT_SHIFT,
BITWISE_AND,
@@ -79,8 +76,6 @@
POW,
SUB,
TABLE,
-
- // Elementwise-Unary
ABS,
BITWISE_NOT,
CEIL,
@@ -92,24 +87,16 @@
NEGATE,
RECIPROCAL,
RSQRT,
-
- // Elementwise-Ternary
SELECT,
-
- // Logical
EQUAL,
GREATER,
GREATER_EQUAL,
-
- // Reduction
REDUCE_ANY,
REDUCE_ALL,
REDUCE_MAX,
REDUCE_MIN,
REDUCE_PRODUCT,
REDUCE_SUM,
-
- // Data layout operation
CONCAT,
PAD,
RESHAPE,
@@ -117,28 +104,18 @@
SLICE,
TILE,
TRANSPOSE,
-
- // Gather/scatter operation
GATHER,
SCATTER,
-
- // Image
RESIZE,
-
- // Type conversion
CAST,
RESCALE,
-
- // Data Nodes
CONST,
IDENTITY,
-
- // Custom operations
CUSTOM,
-
- // Control flow operators
COND_IF,
WHILE_LOOP,
+ FFT2D,
+ RFFT2D,
}
union Attribute {
@@ -162,7 +139,8 @@
MatMulAttribute,
FullyConnectedAttribute,
NegateAttribute,
- CustomAttribute
+ CustomAttribute,
+ FFTAttribute,
}
table PoolAttribute {
@@ -288,6 +266,10 @@
implementation_attrs:[ubyte];
}
+table FFTAttribute {
+ inverse: bool;
+}
+
table Version {
_major: int32 = 0;
_minor: int32 = 51;