Searched refs:reduction_info (Results 1 – 3 of 3) sorted by relevance
594 const ReductionCodegenInfo& reduction_info, int64 x_iter_num);605 ReductionCodegenInfo* reduction_info);614 const ReductionCodegenInfo& reduction_info,624 ReductionCodegenInfo* reduction_info,
4220 ReductionCodegenInfo* reduction_info) { in EmitPrologueForReduction() argument4246 reduction_info->GetMutableReductionInputAddresses(); in EmitPrologueForReduction()4254 int num_partial_results = reduction_info->GetNumPartialResults(); in EmitPrologueForReduction()4256 reduction_info->GetMutablePartialResultAddresses(); in EmitPrologueForReduction()4284 reduction_info->GetMutableInitialValues()->push_back(init_ir_value); in EmitPrologueForReduction()4286 auto& mapping_scheme = reduction_info->GetKernelMappingScheme(); in EmitPrologueForReduction()4291 if (reduction_info->IsRowReduction()) { in EmitPrologueForReduction()4313 reduction_info->GetMutableSharedCache()->push_back(shared_cache_per_reduce); in EmitPrologueForReduction()4363 const ReductionCodegenInfo& reduction_info) { in GetUntransposedOutputLinearAddress() argument4365 reduction_info.GetKernelMappingScheme(); in GetUntransposedOutputLinearAddress()[all …]
323 ReductionDimensions reduction_info = in SharedMemoryUsage() local327 if (reduction_info.is_row_reduction) { in SharedMemoryUsage()