1 2 3 4 5
#[version = "0.0.5"] def @main (%data: Tensor[(2, 2, 2, 2), int32], %weight: Tensor[(2, 2, 2, 2), int32]) { let %x = nn.conv2d(%data, %weight, padding=[1, 1, 1, 1], kernel_size=[2, 2]); %x }