I don’t think we have a pass that does such rewrite. Can you send a PR to add that pass separately from the main one (oneDNN integration)? I think it is very useful. I’m also working on another BYOC which only allows one “bias-like” tensor in a post op.