aboutsummaryrefslogtreecommitdiff
path: root/schema/tosa.fbs
diff options
context:
space:
mode:
Diffstat (limited to 'schema/tosa.fbs')
-rw-r--r--schema/tosa.fbs37
1 files changed, 17 insertions, 20 deletions
diff --git a/schema/tosa.fbs b/schema/tosa.fbs
index 7830bd2..b342103 100644
--- a/schema/tosa.fbs
+++ b/schema/tosa.fbs
@@ -157,24 +157,33 @@ union Attribute {
WhileLoopAttribute,
TransposeAttribute,
TableAttribute,
+ MatMulAttribute,
+ FullyConnectedAttribute,
+ NegateAttribute
}
table PoolAttribute {
pad: [int32];
kernel: [int32];
stride: [int32];
+ input_zp: int32;
+ output_zp: int32;
}
table ConvAttribute {
pad: [int32];
stride: [int32];
dilation: [int32];
+ input_zp: int32;
+ weight_zp: int32;
}
table TransposeConvAttribute {
out_pad: [int32];
stride: [int32];
output_shape: [int32];
+ input_zp: int32;
+ weight_zp: int32;
}
table PadAttribute {
@@ -253,35 +262,24 @@ table TableAttribute {
table: [int16];
}
-union QuantInfo {
- UnaryQuantInfo,
- ConvQuantInfo,
- MatMulQuantInfo,
- PadQuantInfo,
-}
-
-table UnaryQuantInfo {
- input_zp: int32;
- output_zp: int32;
+table MatMulAttribute {
+ a_zp: int32;
+ b_zp: int32;
}
-table ConvQuantInfo {
+table FullyConnectedAttribute {
input_zp: int32;
weight_zp: int32;
}
-table MatMulQuantInfo {
- a_zp: int32;
- b_zp: int32;
-}
-
-table PadQuantInfo {
- input_zp: int32;
+table NegateAttribute {
+ input1_zp: int32;
+ output_zp: int32;
}
table Version {
_major: int32 = 0;
- _minor: int32 = 25;
+ _minor: int32 = 30;
_patch: int32 = 0;
_draft: bool = true;
}
@@ -298,7 +296,6 @@ table TosaOperator {
attribute: Attribute; // union structure. operator attribute
inputs:[string]; // list of input tensor names
outputs:[string]; // list of output tensor names
- quant_info: QuantInfo; // op-based quantization information
}
table TosaBasicBlock {