Searched refs:create_multi_output_jit_fn (Results 1 – 1 of 1) sorted by relevance
5 from torch.cuda.jiterator import _create_multi_output_jit_fn as create_multi_output_jit_fn unknown145 jitted_fn = create_multi_output_jit_fn(code_string, num_outputs)