# feature format: NCHW
feature = tvm.placeholder((16, 3, 244, 244), name="feature")
# kernel format : output_channel, input_channel, kernel_height, kernel_width
kernel = tvm.placeholder((10, 3, 16, 16), name="kernel")
output_data = topi.nn.conv2d(feature, kernel, 1, 0)
relu_result = topi.nn.elemwise.relu(output_data)
s = tvm.create_schedule(output_data.op)
print(tvm.lower(s, [feature, kernel], simple_mode=True))
Code like this will generate two ‘produce’ blocks, is it possible to fuse them through primitive API?