How can I make a whole network into single task?

Hello.

I have question for AutoScheduler.

I’ve been trying to integrate subgraphs to one graph, so I can make one task for a whole network.

I tried a lot of things, but none of them worked.

Is there a recommended way to make the whole model into one single task, so that AutoScheduler can optimize it as a whole?

Any tips, internal constraints, or code-level hints would be appreciated.

Thank you.