MCPcopy Create free account
hub / github.com/pytorch/pytorch / GetPlanGraph

Function GetPlanGraph

caffe2/python/net_drawer.py:258–261  ·  view source on GitHub ↗
(plan_def, name=None, rankdir='TB')

Source from the content-addressed store, hash-verified

256
257
258def GetPlanGraph(plan_def, name=None, rankdir='TB'):
259 graph = pydot.Dot(name, rankdir=rankdir)
260 _draw_steps(plan_def.execution_step, graph)
261 return graph
262
263
264def GetGraphInJson(operators_or_net, output_filepath):

Callers

nothing calls this directly

Calls 1

_draw_stepsFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…