Searched refs:replace_node_module (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | optimization.py | 45 def replace_node_module(node: fx.Node, modules: Dict[str, Any], new_module: torch.nn.Module): function
78 replace_node_module(node.args[0], modules, fused_conv)
146 replace_node_module(node, modules, new_module)
159 replace_node_module(node, modules, old_modules[cur_module])
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | pre_grad.py | 13 replace_node_module,
441 replace_node_module(bn_node.args[0], modules, fused_conv)
533 replace_node_module(bn_node.args[0], modules, fused_conv)
|