Searched refs:ordered_kwargs_for_cpp_kernel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | conv.py | 551 ordered_kwargs_for_cpp_kernel = [
562 ordered_kwargs_for_cpp_kernel.insert(0, "bias")
575 ordered_kwargs_for_cpp_kernel,
|
| H A D | mm.py | 324 ordered_kwargs_for_cpp_kernel = ("beta", "alpha")
433 ordered_kwargs_for_cpp_kernel,
462 ordered_kwargs_for_cpp_kernel,
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | select_algorithm.py | 830 ordered_kwargs_for_cpp_kernel=(), argument
833 self.ordered_kwargs_for_cpp_kernel = ordered_kwargs_for_cpp_kernel
993 ordered_kwargs_for_cpp_kernel=self.choice.ordered_kwargs_for_cpp_kernel,
|
| H A D | ir.py | 3952 ordered_kwargs_for_cpp_kernel: Iterable[str] = dataclasses.field( variable in ExternKernel
3974 ordered_kwargs_for_cpp_kernel=(), argument
3988 self.ordered_kwargs_for_cpp_kernel = ordered_kwargs_for_cpp_kernel
4025 and not self.ordered_kwargs_for_cpp_kernel
4027 self.ordered_kwargs_for_cpp_kernel = [
4441 for arg_name in self.ordered_kwargs_for_cpp_kernel:
4564 ordered_kwargs_for_cpp_kernel=(), argument
4576 ordered_kwargs_for_cpp_kernel,
4623 ordered_kwargs_for_cpp_kernel=(), argument
4635 ordered_kwargs_for_cpp_kernel,
[all …]
|