Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dir.py5215 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 Dselect_algorithm.py981 inner = ir.FallbackKernel.create(
H A Dutils.py1671 return isinstance(node, ir.FallbackKernel) and node.op_overload in op
H A Dlowering.py1705 wrap_tensors, ir.FallbackKernel.create(kernel, *args, **kwargs)
1881 ir.FallbackKernel.create(aten.native_dropout.default, x, p, train),
H A Dscheduler.py435 (ir.FallbackKernel, ir.MultiOutput),