diff options
author | Dominic Symes <dominic.symes@arm.com> | 2023-08-04 17:47:29 +0100 |
---|---|---|
committer | Dominic Symes <dominic.symes@arm.com> | 2023-08-07 09:43:19 +0100 |
commit | 12ab5da01cbc152ed14f00fccdf94815dd1512d2 (patch) | |
tree | f9240bb3292914e954a8a820167b2b1a4bb79090 /tosa.xml | |
parent | 318ac6fc76b9efabb60f5d9c1abf84508e1a8a01 (diff) | |
download | specification-12ab5da01cbc152ed14f00fccdf94815dd1512d2.tar.gz |
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>
Diffstat (limited to 'tosa.xml')
-rw-r--r-- | tosa.xml | 10 |
1 files changed, 5 insertions, 5 deletions
@@ -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> |