Home
last modified time | relevance | path

Searched refs:_wrapped_methods_to_patch (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Dpatcher.py107 torch.fx._symbolic_trace._wrapped_methods_to_patch
110 torch.fx._symbolic_trace._wrapped_methods_to_patch
120 torch.fx._symbolic_trace._wrapped_methods_to_patch = desired_wrapped_methods
133 torch.fx._symbolic_trace._wrapped_methods_to_patch = (
/aosp_15_r20/external/pytorch/torch/fx/
H A D_symbolic_trace.py923 _wrapped_methods_to_patch: List[Tuple[type, str]] = [] variable
931 _wrapped_methods_to_patch.append((torch.Tensor, "__getitem__"))
1145 for cls, name in _wrapped_methods_to_patch:
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py3933 from torch.fx._symbolic_trace import _wrapped_methods_to_patch
3934 _wrapped_methods_to_patch.append((torch.Tensor, "__getitem__"))
3938 _wrapped_methods_to_patch.pop()