Searched refs:python_kernel_name (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | mkldnn_ir.py | 259 self.python_kernel_name,
334 self.python_kernel_name,
429 self.python_kernel_name,
518 self.python_kernel_name,
717 self.python_kernel_name,
982 self.python_kernel_name,
1118 self.python_kernel_name,
1176 self.python_kernel_name,
1240 self.python_kernel_name,
1420 self.python_kernel_name,
[all …]
|
| H A D | ir.py | 3948 python_kernel_name: Optional[str] = None variable in ExternKernel
3972 python_kernel_name=None, argument
3985 self.python_kernel_name = python_kernel_name
4086 else self.python_kernel_name
4562 python_kernel_name=None, argument
4574 python_kernel_name,
4596 python_kernel_name="aten.randint.low_out",
4621 python_kernel_name=None, argument
4633 python_kernel_name,
4836 self.python_kernel_name = "aten.bernoulli_"
[all …]
|
| H A D | comm_analysis.py | 44 kernel_name = node.python_kernel_name
|
| H A D | select_algorithm.py | 991 python_kernel_name=self.choice.call_name(),
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cpu.py | 1338 python_kernel_name, argument
1361 if python_kernel_name.startswith("aten.scatter_reduce"):
2120 python_kernel_name: str,
2165 python_kernel_name,
2294 python_kernel_name: str,
|
| H A D | wrapper.py | 821 python_kernel_name, argument
827 if python_kernel_name.startswith("aten.scatter_reduce"):
843 python_kernel_name: str,
|