Searched defs:def_kernel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/ | ||
| H A D | rocm_kernel.py | 62 def def_kernel( member in ROCmTemplateKernel |
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ | ||
| H A D | cuda_kernel.py | 99 def def_kernel( member in CUDATemplateKernel |
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ | ||
| H A D | cpp_template_kernel.py | 54 def def_kernel( member in CppTemplateKernel |
| /aosp_15_r20/external/pytorch/torch/_inductor/ | ||
| H A D | select_algorithm.py | 259 def def_kernel(self, *argnames): member in TritonTemplateKernel |