diff --git a/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td b/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td index 8f6c76a3479f2..a18b6dd532a28 100644 --- a/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td +++ b/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td @@ -887,7 +887,7 @@ def Tosa_TableOp : Tosa_InferShapedTypeOp<"table"> { }]; let arguments = (ins - Tosa_Tensor: $input, + Tosa_Tensor: $input1, Tosa_Tensor1D: $table ); @@ -896,7 +896,7 @@ def Tosa_TableOp : Tosa_InferShapedTypeOp<"table"> { ); let assemblyFormat = [{ - $input `,` $table attr-dict `:` `(` type($input) `,` type($table) `)` `->` type($output) + $input1 `,` $table attr-dict `:` `(` type($input1) `,` type($table) `)` `->` type($output) }]; let hasVerifier = 1; @@ -1664,7 +1664,7 @@ def Tosa_ReverseOp: Tosa_Op<"reverse", [ }]; let arguments = (ins - Tosa_Tensor:$input, + Tosa_Tensor:$input1, I32Attr:$axis ); @@ -1691,7 +1691,7 @@ def Tosa_SliceOp : Tosa_InferShapedTypeOp<"slice"> { }]; let arguments = (ins - Tosa_Tensor:$input, + Tosa_Tensor:$input1, DenseI64ArrayAttr:$start, DenseI64ArrayAttr:$size ); diff --git a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp index e5dc22f4ee293..e773fa821a2f1 100644 --- a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp +++ b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp @@ -1957,7 +1957,7 @@ class ReverseConverter : public OpConversionPattern { matchAndRewrite(tosa::ReverseOp op, OpAdaptor operands, ConversionPatternRewriter &rewriter) const final { auto loc = op.getLoc(); - Value input = operands.getInput(); + Value input = operands.getInput1(); auto inputTy = cast(input.getType()); auto resultTy = cast_or_null(getTypeConverter()->convertType(op.getType())); @@ -2300,7 +2300,7 @@ class TableConverter : public OpConversionPattern { matchAndRewrite(tosa::TableOp op, OpAdaptor operands, ConversionPatternRewriter &rewriter) const final { auto loc = op.getLoc(); - Value input = operands.getInput(); + Value input = operands.getInput1(); Value table = operands.getTable(); auto inputTy = cast(input.getType()); auto tableTy = cast(table.getType()); diff --git a/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp b/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp index c0c015ab34aab..3f104ed1e3f7f 100644 --- a/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp +++ b/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp @@ -144,7 +144,7 @@ TensorType inferReshapeCollapsedType(TensorType lhsType, TensorType rhsType) { for (; currRhsDim < rhsShape.size(); currRhsDim++) { assert(rhsShape[currRhsDim] == 1); } - + return lhsType.clone(intermediateShape); } @@ -264,7 +264,7 @@ class SliceConverter : public OpConversionPattern { matchAndRewrite(tosa::SliceOp sliceOp, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const final { Location loc = sliceOp.getLoc(); - Value input = adaptor.getInput(); + Value input = adaptor.getInput1(); ShapedType resultType = cast(sliceOp.getType()); if (llvm::isa(resultType)) return failure(); diff --git a/mlir/lib/Dialect/Tosa/IR/TosaCanonicalizations.cpp b/mlir/lib/Dialect/Tosa/IR/TosaCanonicalizations.cpp index 56e74cb178ab4..5e5c39507057a 100644 --- a/mlir/lib/Dialect/Tosa/IR/TosaCanonicalizations.cpp +++ b/mlir/lib/Dialect/Tosa/IR/TosaCanonicalizations.cpp @@ -556,7 +556,7 @@ struct ConcatSliceOptimization : public OpRewritePattern { LogicalResult matchAndRewrite(tosa::SliceOp sliceOp, PatternRewriter &rewriter) const override { - Value sliceInput = sliceOp.getInput(); + Value sliceInput = sliceOp.getInput1(); auto concatOp = sliceInput.getDefiningOp(); if (!concatOp) return rewriter.notifyMatchFailure( @@ -1212,11 +1212,11 @@ OpFoldResult ResizeOp::fold(FoldAdaptor adaptor) { } OpFoldResult ReverseOp::fold(FoldAdaptor adaptor) { - auto operand = getInput(); + auto operand = getInput1(); auto operandTy = llvm::cast(operand.getType()); auto axis = getAxis(); auto operandAttr = - llvm::dyn_cast_if_present(adaptor.getInput()); + llvm::dyn_cast_if_present(adaptor.getInput1()); if (operandAttr) return operandAttr; @@ -1229,16 +1229,16 @@ OpFoldResult ReverseOp::fold(FoldAdaptor adaptor) { } OpFoldResult SliceOp::fold(FoldAdaptor adaptor) { - auto inputTy = llvm::dyn_cast(getInput().getType()); + auto inputTy = llvm::dyn_cast(getInput1().getType()); auto outputTy = llvm::dyn_cast(getType()); if (!inputTy || !outputTy) return {}; if (inputTy == outputTy && inputTy.hasStaticShape()) - return getInput(); + return getInput1(); - if (!adaptor.getInput()) + if (!adaptor.getInput1()) return {}; // Cannot create an ElementsAttr from non-int/float/index types @@ -1246,7 +1246,7 @@ OpFoldResult SliceOp::fold(FoldAdaptor adaptor) { !outputTy.getElementType().isIntOrIndexOrFloat()) return {}; - auto operand = llvm::cast(adaptor.getInput()); + auto operand = llvm::cast(adaptor.getInput1()); if (operand.isSplat() && outputTy.hasStaticShape()) { return SplatElementsAttr::get(outputTy, operand.getSplatValue()); } diff --git a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp index 766dca5a81840..7a671907eb496 100644 --- a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp +++ b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp @@ -885,7 +885,7 @@ LogicalResult tosa::SliceOp::inferReturnTypeComponents( } LogicalResult tosa::SliceOp::verify() { - auto inputType = llvm::dyn_cast(getInput().getType()); + auto inputType = llvm::dyn_cast(getInput1().getType()); auto outputType = llvm::dyn_cast(getType()); if (!inputType || !outputType) return success(); @@ -931,7 +931,7 @@ LogicalResult tosa::TableOp::inferReturnTypeComponents( MLIRContext *context, ::std::optional location, TableOp::Adaptor adaptor, SmallVectorImpl &inferredReturnShapes) { - ShapeAdaptor inputShape(adaptor.getInput().getType()); + ShapeAdaptor inputShape(adaptor.getInput1().getType()); if (!inputShape.hasRank()) { inferredReturnShapes.push_back(ShapedTypeComponents()); @@ -944,7 +944,7 @@ LogicalResult tosa::TableOp::inferReturnTypeComponents( } LogicalResult tosa::TableOp::verify() { - TensorType inputType = getInput().getType(); + TensorType inputType = getInput1().getType(); TensorType outputType = getOutput().getType(); if (inputType.hasRank() && outputType.hasRank() && @@ -2052,7 +2052,7 @@ void IfOp::print(OpAsmPrinter &p) { } LogicalResult ReverseOp::verify() { - TensorType inputType = getInput().getType(); + TensorType inputType = getInput1().getType(); TensorType outputType = getOutput().getType(); int32_t reverseAxis = getAxis(); diff --git a/mlir/lib/Dialect/Tosa/Transforms/TosaFolders.cpp b/mlir/lib/Dialect/Tosa/Transforms/TosaFolders.cpp index c76818ad6d3f4..63594d83ccc11 100644 --- a/mlir/lib/Dialect/Tosa/Transforms/TosaFolders.cpp +++ b/mlir/lib/Dialect/Tosa/Transforms/TosaFolders.cpp @@ -1798,7 +1798,7 @@ struct TosaFoldConstantSlice : public TosaFoldConstantBase { return failure(); auto start = op.getStart(); - auto input = op.getInput(); + auto input = op.getInput1(); ElementsAttr inputValues; if (!matchPattern(input, m_Constant(&inputValues))) return failure();