diff options
author | Luke Hutton <luke.hutton@arm.com> | 2023-01-12 22:20:53 +0000 |
---|---|---|
committer | Luke Hutton <luke.hutton@arm.com> | 2023-01-13 10:34:25 +0000 |
commit | 5e268097917825ddaa00a86ee95a4a6c4f50124b (patch) | |
tree | 8cc0f3b0b7a1d8e68e192a4e243020dc69791c5f /schema | |
parent | ca7ce0e94b3ee7339f31b47baa3a3fb4522243a2 (diff) | |
download | serialization_lib-5e268097917825ddaa00a86ee95a4a6c4f50124b.tar.gz |
Add FFT2d and RFFT2d serialization support
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: I79b3684ae1cf24e71ec7098ed7db5be5bc2f6fb2
Diffstat (limited to 'schema')
-rw-r--r-- | schema/tosa.fbs | 40 |
1 files changed, 11 insertions, 29 deletions
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_identifier "TOSA"; // 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 ResizeMode:uint32 { enum Op:uint32 { UNKNOWN = 0, - - // Tensor Operator ARGMAX, AVG_POOL2D, CONV2D, @@ -54,14 +55,10 @@ enum Op:uint32 { MATMUL, MAX_POOL2D, TRANSPOSE_CONV2D, - - // Activation CLAMP, RESERVED, SIGMOID, TANH, - - // Elementwise-Binary ADD, ARITHMETIC_RIGHT_SHIFT, BITWISE_AND, @@ -79,8 +76,6 @@ enum Op:uint32 { POW, SUB, TABLE, - - // Elementwise-Unary ABS, BITWISE_NOT, CEIL, @@ -92,24 +87,16 @@ enum Op:uint32 { 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 @@ enum Op:uint32 { 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 @@ union Attribute { MatMulAttribute, FullyConnectedAttribute, NegateAttribute, - CustomAttribute + CustomAttribute, + FFTAttribute, } table PoolAttribute { @@ -288,6 +266,10 @@ table CustomAttribute { implementation_attrs:[ubyte]; } +table FFTAttribute { + inverse: bool; +} + table Version { _major: int32 = 0; _minor: int32 = 51; |