From 4881c29247d4b411de446b13d9bd58ea93737aac Mon Sep 17 00:00:00 2001
From: Eric Kunze <eric.kunze@arm.com>
Date: Wed, 1 Nov 2023 16:12:07 -0700
Subject: 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
---
 include/attribute.def | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

(limited to 'include/attribute.def')

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)
 
-- 
cgit v1.2.1

