Home
last modified time | relevance | path

Searched refs:CppTemplateKernel (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_micro_gemm.py16 from .cpp_template_kernel import CppTemplateKernel
119 def codegen_define(self, kernel: CppTemplateKernel) -> str: argument
124 kernel: CppTemplateKernel, argument
163 kernel: CppTemplateKernel, argument
169 kernel: CppTemplateKernel, argument
265 def codegen_define(self, kernel: CppTemplateKernel) -> str: argument
450 def codegen_define(self, kernel: CppTemplateKernel) -> str: argument
689 def codegen_define(self, kernel: CppTemplateKernel) -> str: argument
721 kernel: CppTemplateKernel, argument
727 kernel: CppTemplateKernel, argument
H A Dcpp_template.py17 from .cpp_template_kernel import CppTemplateCaller, CppTemplateKernel
45 ), patch.object(ir.FlexibleLayout, "allow_indexing", True), CppTemplateKernel(
88 kernel = CppTemplateKernel(
H A Dcpp_gemm_template.py21 from .cpp_template_kernel import CppTemplateKernel
768 kernel: CppTemplateKernel, argument
H A Dcpp_template_kernel.py40 class CppTemplateKernel(CppKernel): class