Essentially I’m trying to ‘partition’ a TE (that is, the graph of tensors/ComputeOps). The reason I want to do this is because the graph is very large (actually it’s a whole network lowered to TE) and I want to try some alternative scheduling options on only a small subgraph at a time. I can actually do this by only scheduling the ops in my subgraph (and leaving the rest to ‘default’ scheduling), but then I need to lower the entire graph for every small change in scheduling rather than just lowering the subgraph.