Searched defs:generate_user_defined_triton_kernel (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ | ||
| H A D | cpp_wrapper_cuda.py | 216 def generate_user_defined_triton_kernel( member in CppWrapperCuda |
| H A D | wrapper.py | 790 def generate_user_defined_triton_kernel( member in WrapperCodeGen |