Lines Matching defs:contractOp

333 mlir::vector::castAwayContractionLeadingOneDim(vector::ContractionOp contractOp,
336   VectorType oldAccType = dyn_cast<VectorType>(contractOp.getAccType());
347   auto oldIndexingMaps = contractOp.getIndexingMapsArray();
350   auto oldIteratorTypes = contractOp.getIteratorTypes();
366   SmallVector<Value> operands = {contractOp.getLhs(), contractOp.getRhs(),
367                                  contractOp.getAcc()};
369   auto loc = contractOp.getLoc();
420                              contractOp.getContext());
444                                              contractOp.getContext()));
458       rewriter.getArrayAttr(newIteratorTypes), contractOp.getKind());
468       .create<vector::BroadcastOp>(loc, contractOp->getResultTypes()[0],
484   matchAndRewriteMaskableOp(vector::ContractionOp contractOp,
487     return castAwayContractionLeadingOneDim(contractOp, maskingOp, rewriter);