Home
last modified time | relevance | path

Searched refs:get_reduction_code_buffer (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp.py2046 def get_reduction_code_buffer(loops, buffer="prefix"): function
2074 reduction_prefix = get_reduction_code_buffer(loops)
2082 ) = get_reduction_code_buffer(loops, "local")
2098 code.splice(get_reduction_code_buffer(loops, "suffix"))