Searched defs:make_kernel_render (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/ |
| H A D | rocm_template_buffer.py | 10 make_kernel_render, argument
|
| H A D | rocm_template.py | 106 def make_kernel_render( function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_template.py | 83 def make_kernel_render( function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | cuda_template.py | 107 def make_kernel_render( function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | select_algorithm.py | 722 def make_kernel_render(out_node): function
845 make_kernel_render, argument
|
| H A D | ir.py | 3519 def __init__(self, layout, inputs, make_kernel_render): argument
3572 make_kernel_render, argument
3703 make_kernel_render, argument
3717 def __init__(self, layout, inputs, make_kernel_render, template, choice): argument
|