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),
|