Searched refs:FallbackKernel (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | ir.py | 5215 class FallbackKernel(ExternKernelAlloc): class 5452 device_set = {FallbackKernel.find_device(None, x) for x in example_output} 5482 assert isinstance(self, FallbackKernel) 5702 class ComplexView(FallbackKernel): 5780 if isinstance(inp, FallbackKernel) 7859 class EffectfulKernel(FallbackKernel): 8212 class _CollectiveKernel(FallbackKernel):
|
H A D | select_algorithm.py | 981 inner = ir.FallbackKernel.create(
|
H A D | utils.py | 1671 return isinstance(node, ir.FallbackKernel) and node.op_overload in op
|
H A D | lowering.py | 1705 wrap_tensors, ir.FallbackKernel.create(kernel, *args, **kwargs) 1881 ir.FallbackKernel.create(aten.native_dropout.default, x, p, train),
|
H A D | scheduler.py | 435 (ir.FallbackKernel, ir.MultiOutput),
|