Hello! i’m new user of TVM.
after finishing tutorial, i make simple network using Relay.
L1 = relay.nn.conv2d(X_input, param_1, channels=6, kernel_size=[3, 3]) L2 = relay.nn.bias_add(L1, param_2) L3 = relay.nn.relu(L2) L4 = relay.nn.conv2d(L3, param_3, channels=6, kernel_size=[3, 3]) L5 = relay.nn.bias_add(L4, param_4) L6 = relay.nn.relu(L5) L7 = relay.nn.softmax(L6, axis=1)
The L1, L4 are same conv2d and when i lower them in Halide IR, generate just one Halide IR code. but i want to modify Halide IR just L4’s conv2d not L1.
so my plan is modify the code where generate Halide IR from graph. but i can’t easily find the code that generate Halide IR in TVM.
If anyone knows about this part, can someone give me a hint ???