diff options
author | Eric Kunze <eric.kunze@arm.com> | 2023-11-01 16:12:07 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-01 16:14:25 -0700 |
commit | 4881c29247d4b411de446b13d9bd58ea93737aac (patch) | |
tree | c1a55620543fd0fff382a9f8c7ceeab14b6aa5db /include/attribute.def | |
parent | 5917fc7a9392da8fd1e8c68b2d00b89709a31584 (diff) | |
download | serialization_lib-4881c29247d4b411de446b13d9bd58ea93737aac.tar.gz |
Add support for local_bound attribute
local_bound is used to determine when fast convolution algorithms
can be used in implementing the operation.
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: I9970a2544e90a620f46ac4d3d01cec90a15710a9
Diffstat (limited to 'include/attribute.def')
-rw-r--r-- | include/attribute.def | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/include/attribute.def b/include/attribute.def index b72f65a..dc7a569 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -34,19 +34,21 @@ DEF_ATTRIBUTE(Pool, 6, int32_t, S, output_zp, DType, S, accum_dtype) -DEF_ATTRIBUTE(Conv, 5, +DEF_ATTRIBUTE(Conv, 6, int32_t, V, pad, int32_t, V, stride, int32_t, V, dilation, int32_t, S, input_zp, - int32_t, S, weight_zp) + int32_t, S, weight_zp, + bool, S, local_bound) -DEF_ATTRIBUTE(TransposeConv, 5, +DEF_ATTRIBUTE(TransposeConv, 6, int32_t, V, out_pad, int32_t, V, stride, int32_t, V, output_shape, int32_t, S, input_zp, - int32_t, S, weight_zp) + int32_t, S, weight_zp, + bool, S, local_bound) DEF_ATTRIBUTE(Pad, 3, int32_t, V, padding, @@ -126,6 +128,10 @@ DEF_ATTRIBUTE(Custom, 3, string, S, config, uint8_t, V, implementation_attrs) -DEF_ATTRIBUTE(FFT, 1, - bool, S, inverse) +DEF_ATTRIBUTE(FFT, 2, + bool, S, inverse, + bool, S, local_bound) + +DEF_ATTRIBUTE(RFFT, 1, + bool, S, local_bound) |