aboutsummaryrefslogtreecommitdiff
path: root/schema
diff options
context:
space:
mode:
authorJames Ward <james.ward@arm.com>2022-08-05 13:48:37 +0100
committerJames Ward <james.ward@arm.com>2022-10-04 13:02:00 +0100
commit485a11d8cb67c8062c632f0987cd31cedbe93d6d (patch)
treec12841ac81114120428418f1c359c3d145610804 /schema
parentc92710d7259558fb0cd9e9b38d0c78da21c6e2d4 (diff)
downloadserialization_lib-485a11d8cb67c8062c632f0987cd31cedbe93d6d.tar.gz
FP16 support in serialization
* Allow serialization of fp16 data * Add package to support integrated half data-type (half_float::half), independent of native float: http://half.sourceforge.net/ * Allow passing of accumulate data-type in serialization Signed-off-by: James Ward <james.ward@arm.com> Change-Id: I54357f02e3776d81958228f699ea5044f2014f4b
Diffstat (limited to 'schema')
-rw-r--r--schema/tosa.fbs6
1 files changed, 6 insertions, 0 deletions
diff --git a/schema/tosa.fbs b/schema/tosa.fbs
index d6d0f22..b3ab991 100644
--- a/schema/tosa.fbs
+++ b/schema/tosa.fbs
@@ -31,6 +31,7 @@ enum DType:uint32 {
INT48,
FLOAT,
UINT16,
+ FP16,
}
enum ResizeMode:uint32 {
@@ -168,6 +169,7 @@ table PoolAttribute {
stride: [int32];
input_zp: int32;
output_zp: int32;
+ accum_dtype: DType;
}
table ConvAttribute {
@@ -176,6 +178,7 @@ table ConvAttribute {
dilation: [int32];
input_zp: int32;
weight_zp: int32;
+ accum_dtype: DType;
}
table TransposeConvAttribute {
@@ -184,6 +187,7 @@ table TransposeConvAttribute {
output_shape: [int32];
input_zp: int32;
weight_zp: int32;
+ accum_dtype: DType;
}
table PadAttribute {
@@ -262,11 +266,13 @@ table TableAttribute {
table MatMulAttribute {
a_zp: int32;
b_zp: int32;
+ accum_dtype: DType;
}
table FullyConnectedAttribute {
input_zp: int32;
weight_zp: int32;
+ accum_dtype: DType;
}
table NegateAttribute {