Searched refs:RowReductionGetRowsPerWarp (Results 1 – 1 of 1) sorted by relevance
472 int RowReductionGetRowsPerWarp(int reduced_dimension_size) { in RowReductionGetRowsPerWarp() function3626 if (RowReductionGetRowsPerWarp(tiling_scheme.GetDimsInElems()[2]) > in GenerateReductionCodegenState()3863 int num_rows_per_warp = RowReductionGetRowsPerWarp(reduced_dimension_size); in EmitReductionOutputForRowReduction()4467 int rows_per_warp = RowReductionGetRowsPerWarp(dimx); in CalculateVirtualThreadScalingFactorForReduction()4585 if (RowReductionGetRowsPerWarp(reduction_dimensions.dimensions[2]) > 1) { in ComputeReductionCodegenInfo()