Searched refs:assert_functional_graph (Results 1 – 3 of 3) sorted by relevance
23 assert_functional_graph,186 copy_count = assert_functional_graph(fw_module.graph)190 copy_count2 = assert_functional_graph(fw_module.graph)297 assert_functional_graph(fx_g.graph)
403 def assert_functional_graph(fx_g: torch.fx.Graph) -> int: function
43 assert_functional_graph,