Lines Matching defs:auto

63   auto lhsDefOp = op.getLhs().getDefiningOp<LhsExtOp>();
64 auto rhsDefOp = op.getRhs().getDefiningOp<RhsExtOp>();
71 auto lhsInType = cast<VectorType>(lhsDefOp.getIn().getType());
72 auto rhsInType = cast<VectorType>(rhsDefOp.getIn().getType());
137 auto loc = op.getLoc();
138 auto packInputs = [&](Value lhs, Value rhs) {
142 auto lhs = packInputs(op1.getLhs().getDefiningOp()->getOperand(0),
144 auto rhs = packInputs(op1.getRhs().getDefiningOp()->getOperand(0),
153 auto extOp = op.getLhs().getDefiningOp();
158 .Case<arith::ExtFOp>([&](auto) {
163 .Case<arith::ExtSIOp>([&](auto) {
168 .Case<arith::ExtUIOp>([&](auto) {
173 .Default([&](auto) { llvm_unreachable("unexpected extend op!"); });
176 .Case<arith::ExtFOp>([&](auto) {
181 .Case<arith::ExtSIOp>([&](auto) {
186 .Case<arith::ExtUIOp>([&](auto) {
191 .Default([&](auto) { llvm_unreachable("unexpected extend op!"); });
212 auto nxnxv4i32 =
214 auto nxnxv4f32 =
219 auto nxv4i16 = VectorType::get({4}, rewriter.getI16Type(), true);
220 auto nxv4f16 = VectorType::get({4}, rewriter.getF16Type(), true);
221 auto nxv4bf16 = VectorType::get({4}, rewriter.getBF16Type(), true);
257 auto currentOp = outerProductChain.back();
258 auto acc = currentOp.getAcc();
261 auto previousOp = acc.getDefiningOp<arm_sme::OuterProductOp>();
285 auto loc = op.getLoc();
286 auto packInputs = [&](Value lhs, Value rhs) {
290 auto lhs0 = packInputs(op1.getLhs().getDefiningOp()->getOperand(0),
292 auto lhs1 = packInputs(op2.getLhs().getDefiningOp()->getOperand(0),
294 auto lhs = packInputs(lhs0, lhs1);
296 auto rhs0 = packInputs(op1.getRhs().getDefiningOp()->getOperand(0),
298 auto rhs1 = packInputs(op2.getRhs().getDefiningOp()->getOperand(0),
300 auto rhs = packInputs(rhs0, rhs1);
305 auto lhs0Mask = packInputs(op1.getLhsMask(), op3.getLhsMask());
306 auto lhs1Mask = packInputs(op2.getLhsMask(), op4.getLhsMask());
309 auto rhs0Mask = packInputs(op1.getRhsMask(), op3.getRhsMask());
310 auto rhs1Mask = packInputs(op2.getRhsMask(), op4.getRhsMask());
314 auto lhsExtOp = op.getLhs().getDefiningOp();
315 auto rhsExtOp = op.getRhs().getDefiningOp();
384 auto nxnxv4i32 =
386 auto nxnxv2i64 =
392 auto nxv4i8 = VectorType::get({4}, rewriter.getI8Type(), true);
393 auto nxv2i16 = VectorType::get({2}, rewriter.getI16Type(), true);
395 auto failedToMatch = [&](VectorType resultType, VectorType inputType,
396 auto lhsExtendOp, auto rhsExtendOp) {
399 for (auto op : ops) {
443 auto numScalableDims = resultType.getNumScalableDims();
448 auto *extendOp = extractOp.getVector().getDefiningOp();
454 auto loc = extractOp.getLoc();
489 auto *extendOp = extractOp.getSource().getDefiningOp();
495 auto loc = extractOp.getLoc();