How to match the pattern of a function in Relay?

Thanks @mbrookhart @mbaret for the valuable information! Here is a short summary of this issue.

  • The problem OP has should be resolved by simply matching and partitioning patterns before FuseOps.
  • In addition to that, it would still be better to have FunctionPattern that matches and rewrites Relay functions.

I’m interested in adding the node but I may not have bandwidth in these days either. @mbrookhart I could ping you when I get a chance to work on it to avoid duplicated work :slight_smile: