Searched refs:all_reduction_dims (Results 1 – 2 of 2) sorted by relevance
257 all_reduction_dims = set(comprehension.all_reduction_dims)258 if len(all_reduction_dims) != 1:262 self.reduction_dims = next(iter(all_reduction_dims))
663 def all_reduction_dims(self) -> Set[Tuple[DimDef, ...]]: member in Comprehension