diff options
author | Eric Kunze <eric.kunze@arm.com> | 2022-06-07 05:17:37 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2022-06-15 08:30:47 -0700 |
commit | bdcc3fee1b8bf55aac50e060115b92a1ccf9741c (patch) | |
tree | 05951f3fd1b9e42f859e999976715e6e673acb15 /include/attribute.def | |
parent | a336d54aca08b06953a8b6c49d7e5f6c4899952e (diff) | |
download | serialization_lib-bdcc3fee1b8bf55aac50e060115b92a1ccf9741c.tar.gz |
Remove quantinfo types
Any needed information has been moved into the attributes for each operator.
This aligns with the structure of the attributes in the TOSA
specification, and generally simplifies the code.
Change-Id: I8243e91b09de1a9115f8af09c5e7def7e8f2866b
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Diffstat (limited to 'include/attribute.def')
-rw-r--r-- | include/attribute.def | 29 |
1 files changed, 23 insertions, 6 deletions
diff --git a/include/attribute.def b/include/attribute.def index 8ac5d8d..ea91869 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -26,20 +26,26 @@ ...: variadic variables for more arguments, depending on NUM_ARGS_IN_ATTRIBUTES */ -DEF_ATTRIBUTE(Pool, 3, +DEF_ATTRIBUTE(Pool, 5, int32_t, V, pad, int32_t, V, kernel, - int32_t, V, stride) + int32_t, V, stride, + int32_t, S, input_zp, + int32_t, S, output_zp) -DEF_ATTRIBUTE(Conv, 3, +DEF_ATTRIBUTE(Conv, 5, int32_t, V, pad, int32_t, V, stride, - int32_t, V, dilation) + int32_t, V, dilation, + int32_t, S, input_zp, + int32_t, S, weight_zp) -DEF_ATTRIBUTE(TransposeConv, 3, +DEF_ATTRIBUTE(TransposeConv, 5, int32_t, V, out_pad, int32_t, V, stride, - int32_t, V, output_shape) + int32_t, V, output_shape, + int32_t, S, input_zp, + int32_t, S, weight_zp) DEF_ATTRIBUTE(Pad, 3, int32_t, V, padding, @@ -103,3 +109,14 @@ DEF_ATTRIBUTE(Transpose, 1, DEF_ATTRIBUTE(Table, 1, int16_t, V, table) +DEF_ATTRIBUTE(MatMul, 2, + int32_t, S, a_zp, + int32_t, S, b_zp) + +DEF_ATTRIBUTE(FullyConnected, 2, + int32_t, S, input_zp, + int32_t, S, weight_zp) + +DEF_ATTRIBUTE(Negate, 2, + int32_t, S, input1_zp, + int32_t, S, output_zp) |