Skip to content
Commit a641dd55 authored by Tai Ly's avatar Tai Ly
Browse files

Add support for bias broadcasting



add support for bias broadcasting for operators:
  - conv2d
  - conv3d
  - depthwise_conv2d
  - transpose_conv2d
  - fully_connected

could not add framework test for this because tf.nn.bias_add requires
bias size to match channel dimension.

manually tested reference model evaluation on tosa mlir with bias size
of 1

Signed-off-by: default avatarTai Ly <tai.ly@arm.com>
Change-Id: I70d29d231a63fc03b10e3006cbd6b16b53cca1f2
parent 898d3a20
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment