From 12ab5da01cbc152ed14f00fccdf94815dd1512d2 Mon Sep 17 00:00:00 2001
From: Dominic Symes <dominic.symes@arm.com>
Date: Fri, 4 Aug 2023 17:47:29 +0100
Subject: Allow bias tensors to be broadcast

Bias tensors often have the same value for
each channel. To allow for this the number of
bias channels, BC, is permitted to be 1 or the
output channel size. If BC == 1 then the
bias is broadcast.

Change-Id: I3262a6c699bd045f44201e2a123f9b792b9f67c9
Signed-off-by: Dominic Symes <dominic.symes@arm.com>
---
 tosa.xml | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'tosa.xml')

diff --git a/tosa.xml b/tosa.xml
index cb56914..707ea3b 100644
--- a/tosa.xml
+++ b/tosa.xml
@@ -129,7 +129,7 @@
             <levellimit value="dilation_x * KW" limit="MAX_KERNEL"/>
             <rank min="4" max="4"/>
           </argument>
-          <argument category="input" name="bias" type="tensor_t" shape="[OC]" tensor-element-type="out_t">
+          <argument category="input" name="bias" type="tensor_t" shape="[BC]" tensor-element-type="out_t">
             <description>Per output channel bias data.</description>
             <rank min="1" max="1"/>
           </argument>
@@ -203,7 +203,7 @@
             <levellimit value="dilation_x * KW" limit="MAX_KERNEL"/>
             <rank min="5" max="5"/>
           </argument>
-          <argument category="input" name="bias" type="tensor_t" shape="[OC]" tensor-element-type="out_t">
+          <argument category="input" name="bias" type="tensor_t" shape="[BC]" tensor-element-type="out_t">
             <description>Per output channel bias data.</description>
             <rank min="1" max="1"/>
           </argument>
@@ -279,7 +279,7 @@
             <levellimit value="dilation_x * KW" limit="MAX_KERNEL"/>
             <rank min="4" max="4"/>
           </argument>
-          <argument category="input" name="bias" type="tensor_t" shape="[C*M]" tensor-element-type="out_t">
+          <argument category="input" name="bias" type="tensor_t" shape="[BC]" tensor-element-type="out_t">
             <description>Per output channel bias data.</description>
             <rank min="1" max="1"/>
           </argument>
@@ -384,7 +384,7 @@
             <description>Weights</description>
             <rank min="2" max="2"/>
           </argument>
-          <argument category="input" name="bias" type="tensor_t" shape="[OC]" tensor-element-type="out_t">
+          <argument category="input" name="bias" type="tensor_t" shape="[BC]" tensor-element-type="out_t">
             <description>Per output channel bias data.</description>
             <rank min="1" max="1"/>
           </argument>
@@ -562,7 +562,7 @@
             <levellimit value="KW" limit="MAX_KERNEL"/>
             <rank min="4" max="4"/>
           </argument>
-          <argument category="input" name="bias" type="tensor_t" shape="[OC]" tensor-element-type="out_t">
+          <argument category="input" name="bias" type="tensor_t" shape="[BC]" tensor-element-type="out_t">
             <description>Per output channel bias data.</description>
             <rank min="1" max="1"/>
           </argument>
-- 
cgit v1.2.1