| 28 | |
| 29 | def _graph_debug(test_case, v_level=0, ranks=None, max_py_stack_depth=2): |
| 30 | class DebugGraph(flow.nn.Graph): |
| 31 | def __init__(self): |
| 32 | super().__init__() |
| 33 | self.m = flow.nn.Linear(3, 3) |
| 34 | |
| 35 | def build(self, x): |
| 36 | return x |
| 37 | |
| 38 | d_g = DebugGraph() |
| 39 | d_g.debug(v_level, ranks=ranks, max_py_stack_depth=max_py_stack_depth) |