Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_autodiff_subgraph_slicing.py24 disable_autodiff_subgraph_inlining,
44 with disable_autodiff_subgraph_inlining():
72 with disable_autodiff_subgraph_inlining():
/aosp_15_r20/external/pytorch/test/
H A Dtest_ops_jit.py32 disable_autodiff_subgraph_inlining,
137 with disable_autodiff_subgraph_inlining():
H A Dtest_jit.py103 …ing._internal.jit_utils import JitTestCase, enable_cpu_fuser, disable_autodiff_subgraph_inlining, \
1796 with disable_autodiff_subgraph_inlining():
1878 with disable_autodiff_subgraph_inlining():
5327 with disable_autodiff_subgraph_inlining():
5371 with disable_autodiff_subgraph_inlining():
5394 with disable_autodiff_subgraph_inlining():
5415 with disable_autodiff_subgraph_inlining():
7547 with disable_autodiff_subgraph_inlining():
10995 with disable_autodiff_subgraph_inlining():
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Djit_utils.py673 def disable_autodiff_subgraph_inlining(enabled=True): function