Searched refs:reduction_output_shape_indices (Results 1 – 1 of 1) sorted by relevance
2654 InlinedVector<ShapeIndex, 1>* reduction_output_shape_indices = in EmitPrologueForOneReduction() local2656 reduction_output_shape_indices->push_back(std::move(output_shape_index)); in EmitPrologueForOneReduction()2787 absl::Span<const ShapeIndex> reduction_output_shape_indices = in EmitEpilogueForReduction() local2822 reduction_output_shape_indices[i]), in EmitEpilogueForReduction()2826 reduction_output_shape_indices[i]) in EmitEpilogueForReduction()