[RFC] Visualizing Relay program as graph

Development pre-RFC
24

any way to visualize optimized relay IR? this seems not working to visualize optimized relay IR.

opt_level = 4
target = tvm.target.Target('cuda')
with tvm.transform.PassContext(opt_level=opt_level):
    mod,params = tvm.relay.optimize(mod,target,params=params)
# visualize mod['main']
1 Like