@@ -665,29 +665,27 @@ struct TensorReshapeRewriter : public OpRewritePattern<tensor::ReshapeOp> {
665
665
srcDcvs.push_back (srcLcvs[lvl]);
666
666
}
667
667
668
- Value collapsed_size = constantIndex (builder, loc, 1 );
668
+ Value collapseSize = constantIndex (builder, loc, 1 );
669
669
for (Dimension d = 0 ; d < srcRank; d++)
670
- collapsed_size =
671
- builder.create <arith::MulIOp>(loc, collapsed_size , srcSizes[d]);
672
- SmallVector<Value, 1 > collapsedSizes = {collapsed_size };
670
+ collapseSize =
671
+ builder.create <arith::MulIOp>(loc, collapseSize , srcSizes[d]);
672
+ SmallVector<Value, 1 > collapsedSizes = {collapseSize };
673
673
674
- ReassociationIndices collapse_indices ;
674
+ ReassociationIndices collapseIdx ;
675
675
for (Dimension i = 0 ; i < srcRank; i++)
676
- collapse_indices.push_back (i);
677
- SmallVector<ReassociationIndices, 1 > collapse_reassociation = {
678
- collapse_indices};
676
+ collapseIdx.push_back (i);
677
+ SmallVector<ReassociationIndices, 1 > collapseReass = {collapseIdx};
679
678
SmallVector<Value, 1 > collapsedDcvs;
680
- reshapeCvs (builder, loc, collapse_reassociation , srcSizes, srcDcvs,
679
+ reshapeCvs (builder, loc, collapseReass , srcSizes, srcDcvs,
681
680
collapsedSizes, collapsedDcvs);
682
681
683
- ReassociationIndices expand_indices ;
682
+ ReassociationIndices expandIdx ;
684
683
for (Dimension i = 0 ; i < dstTp.getDimRank (); i++)
685
- expand_indices.push_back (i);
686
- SmallVector<ReassociationIndices, 1 > expand_reassociation = {
687
- expand_indices};
684
+ expandIdx.push_back (i);
685
+ SmallVector<ReassociationIndices, 1 > expandReass = {expandIdx};
688
686
SmallVector<Value> dstDcvs;
689
- reshapeCvs (builder, loc, expand_reassociation , collapsedSizes,
690
- collapsedDcvs, dstSizes, dstDcvs);
687
+ reshapeCvs (builder, loc, expandReass , collapsedSizes, collapsedDcvs ,
688
+ dstSizes, dstDcvs);
691
689
692
690
auto t = builder.create <InsertOp>(loc, v, reduc.front (), dstDcvs);
693
691
builder.create <sparse_tensor::YieldOp>(loc, t);
0 commit comments