Understanding TVM/Relay's PartitionGraph()(mod) function

Questions
4

Isn’t it simply a problem of free variables? I suggest replacing

f = relay.Function([], result)

with

f = relay.Function(relay.analysis.free_vars(result), result)
1 Like