v0.0.4
fn (%data: Tensor[(2, 2, 2, 2), int32], %weight: Tensor[(2, 2, 2, 2), int32]) {
  let %x = nn.conv2d(%data, %weight, channels=2, kernel_size=[2,2]);
  %x
}