Searched defs:reduction_dimensions (Results 1 – 4 of 4) sorted by relevance
78 ReductionDimensions reduction_dimensions = in RewriteReduction() local237 const ReductionDimensions &reduction_dimensions, in RewriteBatchDimensionLargerThanTile()
132 const ReductionDimensions& reduction_dimensions, in GetReductionTiling()194 const ReductionDimensions& reduction_dimensions) { in IsUnnestedReductionFasterThanElemental()664 bool ReductionIsRaceFree(const ReductionDimensions& reduction_dimensions, in ReductionIsRaceFree()
4463 const ReductionDimensions& reduction_dimensions, in CalculateVirtualThreadScalingFactorForReduction()4533 const ReductionDimensions& reduction_dimensions, int num_threads_x, in CanVectorizeReduction()4572 ReductionDimensions reduction_dimensions = in ComputeReductionCodegenInfo() local
4380 DenseIntElementsAttr reduction_dimensions = in matchAndRewrite() local