Searched refs:num_partial_results (Results 1 – 2 of 2) sorted by relevance
172 int num_partial_results, bool is_row_reduction) in ReductionCodegenInfo() argument174 num_partial_results_(num_partial_results), in ReductionCodegenInfo()176 if (num_partial_results > 1) { in ReductionCodegenInfo()177 CHECK_EQ(num_partial_results, (mapping_scheme.GetTileSizeX() / in ReductionCodegenInfo()
4254 int num_partial_results = reduction_info->GetNumPartialResults(); in EmitPrologueForReduction() local4259 element_type, /*ArraySize=*/b_.getInt32(num_partial_results), in EmitPrologueForReduction()4280 for (int i = 0; i < num_partial_results; ++i) { in EmitPrologueForReduction()4295 num_partial_results); in EmitPrologueForReduction()4306 num_partial_results); in EmitPrologueForReduction()4409 int num_partial_results = reduction_info.GetNumPartialResults(); in EmitEpilogueForReduction() local4434 for (int j = 0; j < num_partial_results; ++j) { in EmitEpilogueForReduction()4639 int num_partial_results = reduction_info.GetNumPartialResults(); in EmitTileElementForReduction() local4651 input_gens[i](num_partial_results > 1 ? index_without_linear in EmitTileElementForReduction()4666 /*use_linear_index=*/num_partial_results == 1, extra_output_gens)); in EmitTileElementForReduction()[all …]