Home
last modified time | relevance | path

Searched defs:reduction_hint (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/runtime/
H A Dtriton_heuristics.py1508 reduction_hint=False, argument
1537 reduction_hint=False, argument
1585 reduction_hint=False, argument
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dtriton_split_scan.py35 reduction_hint=torch._inductor.runtime.hints.ReductionHint.DEFAULT, argument
H A Dsimd.py316 reduction_hint=ReductionHint.DEFAULT, argument
1158 def reduction_hint(node): member in SIMDScheduling
H A Dhalide.py677 reduction_hint=ReductionHint.DEFAULT, argument
H A Dtriton.py1203 reduction_hint=ReductionHint.DEFAULT, argument
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dir.py1348 reduction_hint, argument