Lines Matching refs:operands
218 OpFoldResult AnyOp::fold(ArrayRef<Attribute> operands) { in fold() argument
220 if (operands.back()) in fold()
221 return operands.back(); in fold()
238 result.operands)) in parse()
339 Optional<unsigned> index, ArrayRef<Attribute> operands, in getSuccessorRegions() argument
398 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
401 if (operands[0].getType().isa<SizeType>() || in inferReturnTypes()
402 operands[1].getType().isa<SizeType>()) in inferReturnTypes()
414 OpFoldResult mlir::shape::AddOp::fold(ArrayRef<Attribute> operands) { in fold() argument
420 operands, [](APInt a, const APInt &b) { return std::move(a) + b; }); in fold()
444 SmallVector<Value> operands; in matchAndRewrite() local
448 operands.append(assumeAll.operand_begin(), assumeAll->operand_end()); in matchAndRewrite()
450 operands.push_back(operand); in matchAndRewrite()
454 if (operands.size() == op.getNumOperands()) in matchAndRewrite()
458 rewriter.replaceOpWithNewOp<AssumingAllOp>(op, operands); in matchAndRewrite()
490 SetVector<CstrBroadcastableOp> operands; in matchAndRewrite() local
498 operands.insert(broadcastable); in matchAndRewrite()
502 if (operands.size() <= 1) in matchAndRewrite()
507 for (auto cstr : operands) { in matchAndRewrite()
608 OpFoldResult AssumingAllOp::fold(ArrayRef<Attribute> operands) { in fold() argument
611 for (int idx = operands.size() - 1; idx >= 0; idx--) { in fold()
612 Attribute a = operands[idx]; in fold()
641 OpFoldResult BroadcastOp::fold(ArrayRef<Attribute> operands) { in fold() argument
653 if (!operands[0] || !operands[1]) in fold()
656 operands[0].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
658 operands[1].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
851 OpFoldResult ConcatOp::fold(ArrayRef<Attribute> operands) { in fold() argument
852 if (!operands[0] || !operands[1]) in fold()
855 operands[0].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
857 operands[1].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
915 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
970 OpFoldResult CstrBroadcastableOp::fold(ArrayRef<Attribute> operands) { in fold() argument
972 if (hasAtMostSingleNonScalar(operands)) in fold()
977 for (const auto &operand : operands) { in fold()
1022 OpFoldResult CstrEqOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1023 if (llvm::all_of(operands, in fold()
1024 [&](Attribute a) { return a && a == operands[0]; })) in fold()
1064 OpFoldResult CstrRequireOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1065 return operands[0]; in fold()
1072 OpFoldResult DivOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1073 auto lhs = operands[0].dyn_cast_or_null<IntegerAttr>(); in fold()
1076 auto rhs = operands[1].dyn_cast_or_null<IntegerAttr>(); in fold()
1093 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1096 if (operands[0].getType().isa<SizeType>() || in inferReturnTypes()
1097 operands[1].getType().isa<SizeType>()) in inferReturnTypes()
1115 OpFoldResult ShapeEqOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1117 if (!operands.empty() && !operands[0]) in fold()
1119 for (Attribute operand : operands.drop_front(1)) { in fold()
1122 allSame = allSame && operand == operands[0]; in fold()
1131 OpFoldResult IndexToSizeOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1134 if (Attribute arg = operands[0]) in fold()
1148 OpFoldResult FromExtentsOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1149 if (llvm::any_of(operands, [](Attribute a) { return !a; })) in fold()
1152 for (auto attr : operands) in fold()
1245 OpFoldResult GetExtentOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1246 auto elements = operands[0].dyn_cast_or_null<DenseIntElementsAttr>(); in fold()
1272 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1296 OpFoldResult IsBroadcastableOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1298 if (operands.size() < 2) { in fold()
1310 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1313 inferredReturnTypes.assign({operands[0].getType()}); in inferReturnTypes()
1341 OpFoldResult shape::RankOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1342 auto shape = operands[0].dyn_cast_or_null<DenseIntElementsAttr>(); in fold()
1398 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1401 if (operands[0].getType().isa<ShapeType>()) in inferReturnTypes()
1419 OpFoldResult NumElementsOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1422 Attribute shape = operands[0]; in fold()
1434 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1437 if (operands[0].getType().isa<ShapeType>()) in inferReturnTypes()
1458 OpFoldResult MaxOp::fold(llvm::ArrayRef<mlir::Attribute> operands) { in fold() argument
1466 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1469 if (operands[0].getType() == operands[1].getType()) in inferReturnTypes()
1470 inferredReturnTypes.assign({operands[0].getType()}); in inferReturnTypes()
1490 OpFoldResult MinOp::fold(llvm::ArrayRef<mlir::Attribute> operands) { in fold() argument
1498 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1501 if (operands[0].getType() == operands[1].getType()) in inferReturnTypes()
1502 inferredReturnTypes.assign({operands[0].getType()}); in inferReturnTypes()
1522 OpFoldResult MulOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1523 auto lhs = operands[0].dyn_cast_or_null<IntegerAttr>(); in fold()
1526 auto rhs = operands[1].dyn_cast_or_null<IntegerAttr>(); in fold()
1535 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1538 if (operands[0].getType().isa<SizeType>() || in inferReturnTypes()
1539 operands[1].getType().isa<SizeType>()) in inferReturnTypes()
1622 MLIRContext *context, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1625 if (operands[0].getType().isa<ValueShapeType>()) in inferReturnTypes()
1628 auto shapedTy = operands[0].getType().cast<ShapedType>(); in inferReturnTypes()
1667 OpFoldResult SizeToIndexOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1670 if (Attribute arg = operands[0]) in fold()
1693 auto operands = getOperands(); in verify() local
1698 for (auto e : llvm::zip(results, operands)) in verify()
1709 LogicalResult SplitAtOp::fold(ArrayRef<Attribute> operands, in fold() argument
1711 if (!operands[0] || !operands[1]) in fold()
1714 operands[0].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
1716 auto splitPoint = operands[1].cast<IntegerAttr>().getInt(); in fold()
1724 Builder builder(operands[0].getContext()); in fold()
1734 OpFoldResult ToExtentTensorOp::fold(ArrayRef<Attribute> operands) { in fold() argument
1735 if (!operands[0]) in fold()
1739 operands[0].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
1828 SmallVector<OpAsmParser::UnresolvedOperand, 3> operands; in parse() local
1830 if (parser.parseOperandList(operands, /*requiredOperandCount=*/-1, in parse()
1837 auto initVals = llvm::makeArrayRef(operands).drop_front(); in parse()
1838 if (parser.resolveOperand(operands.front(), shapeOrExtentTensorType, in parse()
1839 result.operands) || in parse()
1841 result.operands)) in parse()