Searched defs:define_kernel (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/ |
| H A D | rocm_cpp_scheduling.py | 44 def define_kernel(self, src_code: str, node_schedule) -> str: member in ROCmCPPScheduling
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | cuda_cpp_scheduling.py | 49 def define_kernel(self, src_code: str, node_schedule) -> str: member in CUDACPPScheduling
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | multi_kernel.py | 88 def define_kernel(self, kernels): member in MultiKernelState
|
| H A D | cpp_wrapper_cuda.py | 197 def define_kernel( member in CppWrapperCuda
|
| H A D | simd.py | 1821 def define_kernel(self, src_code, node_schedule, kernel): member in SIMDScheduling
|
| H A D | cpp_wrapper_cpu.py | 923 def define_kernel( member in CppWrapperCpu
|
| H A D | halide.py | 1672 def define_kernel(self, src_code, node_schedule, kernel): member in HalideScheduling
|
| H A D | wrapper.py | 1251 def define_kernel( member in WrapperCodeGen
|
| H A D | triton.py | 3011 def define_kernel(self, src_code, node_schedule, kernel): member in TritonScheduling
|
| H A D | cpp.py | 4544 def define_kernel(self, src_code, nodes, kernel_args=None): member in CppScheduling
|