We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dfb5b6e commit 9e65034Copy full SHA for 9e65034
mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
@@ -483,8 +483,7 @@ LogicalResult tosa::ArgMaxOp::verify() {
483
484
const ArrayRef<int64_t> inputShape = inputType.getShape();
485
const ArrayRef<int64_t> outputShape = resultType.getShape();
486
- llvm::SmallVector<int64_t> expectedOutputShape(inputShape.begin(),
487
- inputShape.end());
+ llvm::SmallVector<int64_t> expectedOutputShape(inputShape);
488
expectedOutputShape.erase(expectedOutputShape.begin() + axis);
489
if (failed(verifyCompatibleShape(expectedOutputShape, outputShape)))
490
return emitOpError("expected output shape '")
0 commit comments