Searched refs:GetReductionOutputShapeIndices (Results 1 – 1 of 1) sorted by relevance
2571 absl::Span<const ShapeIndex> GetReductionOutputShapeIndices() const { in GetReductionOutputShapeIndices() function in xla::gpu::ReductionCodegenInfo2788 reduction_info->GetReductionOutputShapeIndices(); in EmitEpilogueForReduction()