Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_executorch_gen.py12 ComputeCodegenUnboxedKernels,
495 result = ComputeCodegenUnboxedKernels(selector, use_aten_lib)(entry)
540 Exception, ComputeCodegenUnboxedKernels(selector, use_aten_lib), entry
562 result = ComputeCodegenUnboxedKernels(selector, use_aten_lib)(entry)
583 result = ComputeCodegenUnboxedKernels(selector, use_aten_lib)(entry)
619 result = ComputeCodegenUnboxedKernels(selector, use_aten_lib)(entry)
/aosp_15_r20/external/pytorch/tools/jit/
H A Dgen_unboxing.py95 class ComputeCodegenUnboxedKernels: class
197 "unboxed_ops": [ComputeCodegenUnboxedKernels(selector)(fn)]
/aosp_15_r20/external/pytorch/torchgen/
H A Dgen_executorch.py179 class ComputeCodegenUnboxedKernels: class
320 ComputeCodegenUnboxedKernels(selector, use_aten_lib)(unbox_kernel_entry)