Searched refs:reduce_graph_module (Results 1 – 3 of 3) sorted by relevance
7 reduce_graph_module,165 return (reduce_graph_module, (dict_without_graph, import_block))
123 def reduce_graph_module(body: Dict[Any, Any], import_block: str) -> torch.nn.Module: function829 return (reduce_graph_module, (dict_without_graph, import_block))
33 torch.fx.graph_module.reduce_graph_module(body: Dict[Any, Any], import_block: str) -> torch.nn.modu…