Searched refs:reduction_info (Results 1 – 1 of 1) sorted by relevance
2646 ReductionCodegenInfo* reduction_info = in EmitPrologueForOneReduction() local2650 reduction_info->GetMutableReducers(); in EmitPrologueForOneReduction()2655 reduction_info->GetMutableReductionOutputShapeIndices(); in EmitPrologueForOneReduction()2659 reduction_info->GetMutableReductionInputAddresses(); in EmitPrologueForOneReduction()2665 int num_partial_results = reduction_info->GetNumberOfPartialResults(); in EmitPrologueForOneReduction()2667 reduction_info->GetMutablePartialResultAddresses(); in EmitPrologueForOneReduction()2706 ReductionCodegenInfo* reduction_info = in EmitPrologueForReduction() local2732 int num_partial_results = reduction_info->GetNumberOfPartialResults(); in EmitPrologueForReduction()2736 reduction_info->SetCurrentOutputLinearIndexAddress( in EmitPrologueForReduction()2737 Alloca(reduction_info->GetIndexType(), in EmitPrologueForReduction()[all …]