Home
last modified time | relevance | path

Searched refs:annotate_getitem_nodes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/passes/
H A Dannotate_getitem_nodes.py6 def annotate_getitem_nodes(graph: torch.fx.Graph) -> None: function
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx_experimental.py45 from torch.fx.passes.annotate_getitem_nodes import annotate_getitem_nodes
1164 annotate_getitem_nodes(my_module_traced.graph)
/aosp_15_r20/external/pytorch/
H A D.lintrunner.toml1281 'torch/fx/passes/annotate_getitem_nodes.py',
/aosp_15_r20/external/pytorch/docs/source/
H A Dfx.rst1171 .. py:module:: torch.fx.passes.annotate_getitem_nodes