Pass the context to the perform method of nodes that have a context.
Sweep all the ops using the default context and change them to either
use the context of one of the inputs or take it as a parameter.