hello.
I’m trying to apply my custom ir pass to Conv2D layer from network. As i learned in this tutorial, i created a custom ir pass function called ModifyConv2D. And to do this, I write the code below.
The problem is that ModifyConv2D needs to find only the conv layer and modify it, but i can’t find a solution that detect conv layer and apply custiom ir pass.
Some people told me to use autotvm.task and I used the following code but still couldn’t find a way.
tasks = autotvm.task.extract_from_program(func , target='cuda' , params=params, ops = (relay.op.nn.conv2d. ) )
is there any examples or solutions for this problem?