Home
last modified time | relevance | path

Searched refs:script_if_tracing (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/jit/
H A D__init__.py176 def script_if_tracing(fn): function
/aosp_15_r20/external/pytorch/torch/distributions/
H A Dvon_mises.py90 @torch.jit.script_if_tracing
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_functions.py93 @torch.jit.script_if_tracing
113 @torch.jit.script_if_tracing
/aosp_15_r20/external/pytorch/docs/source/
H A Djit.rst56 script_if_tracing
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py3512 untraceable = torch.jit.script_if_tracing(untraceable)
3525 @torch.jit.script_if_tracing