Searched defs:reduction_ranges (Results 1 – 1 of 1) sorted by relevance
678 reduction_ranges, argument898 def _unroll_reduction_fn(inner_fn, reduction_ranges, reduction_type, src_dtype): argument1133 reduction_ranges, argument1346 reduction_ranges, argument