Home
last modified time | relevance | path

Searched defs:CppTemplateKernel (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_micro_gemm.py119 def codegen_define(self, kernel: CppTemplateKernel) -> str:
124 kernel: CppTemplateKernel,
163 kernel: CppTemplateKernel,
169 kernel: CppTemplateKernel,
265 def codegen_define(self, kernel: CppTemplateKernel) -> str:
450 def codegen_define(self, kernel: CppTemplateKernel) -> str:
689 def codegen_define(self, kernel: CppTemplateKernel) -> str:
721 kernel: CppTemplateKernel,
727 kernel: CppTemplateKernel,
H A Dcpp_gemm_template.py768 kernel: CppTemplateKernel,
H A Dcpp_template_kernel.py40 class CppTemplateKernel(CppKernel): class