v0.0.4
fn (%x: Tensor[(2, 4), int32], %bias: Tensor[(4), int32])  {
  let %x1: Tensor[(2, 4), int32] = nn.bias_add(%x, %bias, axis=1);
  %x1
}