Thanks! I had checked that out, but seems it doesn’t show a way to match a function. In my case conv+mul+add+relu is already wrapped into a function, so I failed to match them directly. One example in the tutorial related to function matching uses function attr, but it looks like the function I have above has a None attr.
Any thoughts?