Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_jiterator.py5 from torch.cuda.jiterator import _create_multi_output_jit_fn as create_multi_output_jit_fn unknown
145 jitted_fn = create_multi_output_jit_fn(code_string, num_outputs)