Searched defs:ReductionHint (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | hints.py | 18 class ReductionHint(Enum): class
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 947 reduction_hint: ReductionHint = ReductionHint.DEFAULT,
1114 split: int, numel_hint: int, reduction_hint: ReductionHint
1203 reduction_hint: ReductionHint,
1263 reduction_hint: ReductionHint,
1303 reduction_hint: ReductionHint,
1389 reduction_hint: ReductionHint = ReductionHint.DEFAULT,
1512 reduction_hint: ReductionHint,
1663 reduction_hint: ReductionHint = ReductionHint.DEFAULT,
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | triton_combo_kernel.py | 471 reduction_hint: ReductionHint,
|