Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dir.py609 reduction_ranges: List[Expr]
625 *(free_unbacked_symbols(e) for e in self.reduction_ranges)
629 return self.reduction_ranges
644 return len(self.ranges) + len(self.reduction_ranges)
648 rindex = self._index(self.reduction_ranges, SymT.RINDEX)
653 rindex = self._index(self.reduction_ranges, SymT.RINDEX)
665 self.reduction_ranges,
678 reduction_ranges, argument
809 reduction_ranges,
829 reduction_ranges,
[all …]
H A Dlowering.py5126 reduction_ranges=reduced_sizes,
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dsimd.py1608 ranges, reduction_ranges = node.get_ranges()
1667 score - sympy_product(itertools.chain(ranges, reduction_ranges))