Searched refs:maybe_replace_node (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
| H A D | duplicate_dq_pass.py | 41 def maybe_replace_node(n: torch.fx.Node) -> torch.fx.Node: function
47 new_args = map_arg(user.args, maybe_replace_node)
48 new_kwargs = map_arg(user.kwargs, maybe_replace_node)
|
| /aosp_15_r20/external/pytorch/torch/fx/ |
| H A D | node.py | 625 def maybe_replace_node(n : Node) -> Node: function
634 new_args = map_arg(use_node.args, maybe_replace_node)
635 new_kwargs = map_arg(use_node.kwargs, maybe_replace_node)
722 def maybe_replace_node(n : Node) -> Node: function
729 new_args = map_arg(self.args, maybe_replace_node)
730 new_kwargs = map_arg(self.kwargs, maybe_replace_node)
|
| /aosp_15_r20/external/executorch/backends/transforms/ |
| H A D | duplicate_dynamic_quant_chain.py | 40 def maybe_replace_node(n: torch.fx.Node) -> torch.fx.Node: function
46 new_args = map_arg(node.args, maybe_replace_node)
47 new_kwargs = map_arg(node.kwargs, maybe_replace_node)
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | quantization.py | 2576 def maybe_replace_node(n: torch.fx.Node) -> torch.fx.Node: function
2582 new_args = map_arg(new_quant_node.args, maybe_replace_node)
2583 new_kwargs = map_arg(new_quant_node.kwargs, maybe_replace_node)
|