diff options
author | Tai Ly <tai.ly@arm.com> | 2024-03-13 18:52:45 +0000 |
---|---|---|
committer | Tai Ly <tai.ly@arm.com> | 2024-03-13 19:25:43 +0000 |
commit | ad78daaf0fa1e41742cbed314459c3dbbb483c20 (patch) | |
tree | f27e56f497b796d36676b42bb713deb1af883a31 /include/attribute.def | |
parent | 0b6d7c271af1e6593e6a2cf14b32acea765f4b64 (diff) | |
download | serialization_lib-ad78daaf0fa1e41742cbed314459c3dbbb483c20.tar.gz |
[serialization_lib] Add acc_type to Conv Attrs
This adds acc_type to ConvAttribute and TransposeConvAttribute
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I73bab71b2eb90f6451fadee21d5bed1811ecbfd7
Diffstat (limited to 'include/attribute.def')
-rw-r--r-- | include/attribute.def | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/include/attribute.def b/include/attribute.def index 2176f47..723543e 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -34,21 +34,23 @@ DEF_ATTRIBUTE(Pool, 6, int32_t, S, output_zp, DType, S, acc_type) -DEF_ATTRIBUTE(Conv, 6, +DEF_ATTRIBUTE(Conv, 7, int32_t, V, pad, int32_t, V, stride, int32_t, V, dilation, int32_t, S, input_zp, int32_t, S, weight_zp, - bool, S, local_bound) + bool, S, local_bound, + DType, S, acc_type) -DEF_ATTRIBUTE(TransposeConv, 6, +DEF_ATTRIBUTE(TransposeConv, 7, int32_t, V, out_pad, int32_t, V, stride, int32_t, V, output_shape, int32_t, S, input_zp, int32_t, S, weight_zp, - bool, S, local_bound) + bool, S, local_bound, + DType, S, acc_type) DEF_ATTRIBUTE(Pad, 1, uint8_t, V, pad_const) |