def test_ofg(self): m = models.Ofg() f = th.function(m.inputs, m.outputs) pdf = PyDotFormatter() graph = pdf(f) assert len(graph.get_nodes()) == 10 sub_graphs = graph.get_subgraph_list() assert len(sub_graphs) == 2 ofg1, ofg2 = sub_graphs assert len(ofg1.get_nodes()) == 5 assert len(ofg1.get_nodes()) == len(ofg2.get_nodes())
def test_ofg(self): m = models.Ofg() f = th.function(m.inputs, m.outputs) self.check(f)