Searched refs:GetNumberOfReduces (Results 1 – 1 of 1) sorted by relevance
2566 int GetNumberOfReduces() const { return reducers_.size(); } in GetNumberOfReduces() function in xla::gpu::ReductionCodegenInfo2782 int num_reduces = reduction_info->GetNumberOfReduces(); in EmitEpilogueForReduction()