v0.0.4 fn (%x: Tensor[(1, 64, 512, 256), float32], %bias: Tensor[(64), float32]) { let %x1: Tensor[(1, 64, 512, 256), float32] = nn.bias_add(%x, %bias, axis=1); %x1 }