@@ -1628,23 +1628,23 @@ Value ConvertAtenConvolutionOp::createTransposedInputPadding(
16281628 Value outerDim = outerSizes[i + 2 ];
16291629 Value isPadding = rewriter.createOrFold <arith::CmpIOp>(
16301630 loc, arith::CmpIPredicate::ugt, outerDim, innerDim);
1631- Value maxDim = rewriter. createOrFold <arith::SelectOp>(loc, isPadding,
1632- outerDim, innerDim);
1631+ Value maxDim =
1632+ rewriter. createOrFold <arith::SelectOp>(loc, isPadding, outerDim, innerDim);
16331633 maxSizes.push_back (maxDim);
16341634 }
16351635
16361636 Value initMaxTensor =
16371637 createInitTensor (rewriter, loc, maxSizes, inputDTy, pad);
16381638
16391639 // Insert input
1640- auto paddedTensor = rewriter. create < tensor::InsertSliceOp>(
1640+ auto paddedTensor = tensor::InsertSliceOp::create (rewriter,
16411641 loc, torch_to_linalg::removeSizeInformation (rewriter, loc, input),
16421642 initMaxTensor, insertSliceOffsets, inputSizes, strideIndexValues);
16431643
16441644 SmallVector<Value> allOnesStrides (inputSizes.size (), c1);
16451645
16461646 // Crop. Extract the final tensor from the "max" tensor
1647- auto finalTensor = rewriter. create < tensor::ExtractSliceOp>(
1647+ auto finalTensor = tensor::ExtractSliceOp::create (rewriter,
16481648 loc,
16491649 torch_to_linalg::removeSizeInformation (rewriter, loc, paddedTensor),
16501650 extractSliceOffsets, outerSizes, allOnesStrides);
@@ -1657,7 +1657,7 @@ Value ConvertAtenConvolutionOp::createTransposedInputPadding(
16571657 createInitTensor (rewriter, loc, outerSizes, inputDTy, pad);
16581658
16591659 // Insert the original input into the outer tensor with calculated offsets
1660- auto paddedInput = rewriter. create < tensor::InsertSliceOp>(
1660+ auto paddedInput = tensor::InsertSliceOp::create (rewriter,
16611661 loc, torch_to_linalg::removeSizeInformation (rewriter, loc, input),
16621662 initPaddedTensor, insertSliceOffsets, inputSizes, strideIndexValues);
16631663 return paddedInput;
0 commit comments