You could

add an op with a subgraph attribute that would run the subgraph in your system,
register a custom pass moving ops you want to run yourself into the subgraph.

This way you can even fall back to running ops you might not support in PyTorch.
Best regards
Thomas