Searched refs:def_kernel (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/_inductor/codegen/rocm/ | ||
| D | rocm_kernel.py | 62 def def_kernel( member in ROCmTemplateKernel |
| D | ck_universal_gemm_template.py | 304 kernel_definition=kernel.def_kernel( |
| /external/pytorch/torch/_inductor/codegen/cuda/ | ||
| D | cuda_kernel.py | 99 def def_kernel( member in CUDATemplateKernel |
| D | gemm_template.py | 914 kernel_call_signature = kernel.def_kernel( |
| /external/pytorch/torch/_inductor/codegen/ | ||
| D | cpp_template_kernel.py | 54 def def_kernel( member in CppTemplateKernel |
| /external/pytorch/torch/_inductor/ | ||
| D | select_algorithm.py | 259 def def_kernel(self, *argnames): member in TritonTemplateKernel |