Searched refs:cpp_kernel_name (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 3949 cpp_kernel_name: Optional[str] = None variable in ExternKernel
3973 cpp_kernel_name=None, argument
3987 self.cpp_kernel_name = cpp_kernel_name or get_aten_cpp_kernel_name(op_overload)
4081 … V.graph.wrapper_code.get_c_shim_func_name(self.cpp_kernel_name) # type: ignore[attr-defined]
4083 else self.cpp_kernel_name
4563 cpp_kernel_name=None, argument
4575 cpp_kernel_name,
4600 cpp_kernel_name="at::_ops::randint_low_out::call"
4622 cpp_kernel_name=None, argument
4634 cpp_kernel_name,
[all …]
|
| H A D | mkldnn_ir.py | 260 self.cpp_kernel_name,
335 self.cpp_kernel_name,
430 self.cpp_kernel_name,
519 self.cpp_kernel_name,
718 self.cpp_kernel_name,
983 self.cpp_kernel_name,
1119 self.cpp_kernel_name,
1177 self.cpp_kernel_name,
1241 self.cpp_kernel_name,
1421 self.cpp_kernel_name,
[all …]
|
| H A D | select_algorithm.py | 799 self.cpp_kernel_name = cpp_kernel
992 cpp_kernel_name=self.choice.cpp_kernel_name,
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cpu.py | 1307 self.generate_c_shim_extern_kernel_call(fallback_kernel.cpp_kernel_name, args)
1337 cpp_kernel_name, argument
1348 cpp_kernel_name = self.get_c_shim_func_name(cpp_kernel_name)
1350 cpp_kernel_name = cpp_kernel_name.replace("__", "_") + "_out"
2121 cpp_kernel_name: str,
2166 cpp_kernel_name,
2295 cpp_kernel_name: str,
|
| H A D | wrapper.py | 820 cpp_kernel_name, argument
844 cpp_kernel_name: str,
|