diff options
Diffstat (limited to 'schema')
-rw-r--r-- | schema/tosa.fbs | 6 |
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 { |