Searched refs:handle_submodule (Results 1 – 1 of 1) sorted by relevance
751 def handle_submodule( function771 handle_submodule(typing.cast(torch.fx.Node, cond_node.args[1]))772 handle_submodule(typing.cast(torch.fx.Node, cond_node.args[2]))775 handle_submodule(typing.cast(torch.fx.Node, while_node.args[0]))776 handle_submodule(typing.cast(torch.fx.Node, while_node.args[1]))780 handle_submodule(