Searched refs:is_row_reduction (Results 1 – 6 of 6) sorted by relevance
99 bool is_row_reduction = reduction_dimensions.is_row_reduction; in RewriteReduction() local102 is_row_reduction ? kRowAtomicFreeBound : kColumnAtomicFreeBound; in RewriteReduction()157 int64 inner_reduced_dimension = is_row_reduction in RewriteReduction()182 int64 outer_reduced_dimension = is_row_reduction in RewriteReduction()207 CHECK(reduction_dimensions.is_row_reduction); in RewriteBatchDimensionLargerThanTile()
172 int num_partial_results, bool is_row_reduction) in ReductionCodegenInfo() argument175 is_row_reduction_(is_row_reduction) { in ReductionCodegenInfo()
176 bool is_row_reduction; member
137 if (reduction_dimensions.is_row_reduction) { in GetReductionTiling()287 if (reduction_dimensions.is_row_reduction) { in IsReductionFromOrToContiguousDimensions()359 if (reduction_dimensions.is_row_reduction) { in IsReductionFromOrToContiguousDimensions()
327 if (reduction_info.is_row_reduction) { in SharedMemoryUsage()
5299 VLOG(10) << "is_row_reduction " << reduction_dimensions.is_row_reduction in ComputeReductionCodegenInfo()5320 int64 num_threads_y = reduction_dimensions.is_row_reduction ? 1 : kWarpSize; in ComputeReductionCodegenInfo()5322 if (reduction_dimensions.is_row_reduction) { in ComputeReductionCodegenInfo()5354 if (reduction_dimensions.is_row_reduction && in ComputeReductionCodegenInfo()5363 } else if (!reduction_dimensions.is_row_reduction && in ComputeReductionCodegenInfo()5392 reduction_dimensions.is_row_reduction); in ComputeReductionCodegenInfo()