From 485a11d8cb67c8062c632f0987cd31cedbe93d6d Mon Sep 17 00:00:00 2001 From: James Ward Date: Fri, 5 Aug 2022 13:48:37 +0100 Subject: 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 Change-Id: I54357f02e3776d81958228f699ea5044f2014f4b --- schema/tosa.fbs | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'schema') 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 { -- cgit v1.2.1