Searched refs:process_kernel (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 4103 def process_kernel( member in ExternKernel
5639 ) = cls.process_kernel(kernel, *args, **kwargs)
8252 ) = cls.process_kernel(kernel, inputs, *args, **kwargs)
8304 ) = cls.process_kernel(kernel, inputs, *args, **kwargs)
8374 ) = cls.process_kernel(kernel, inp)
|
| H A D | scheduler.py | 679 cls.process_kernel(op, *fake_inputs, **self.node.kwargs)
|