Lines Matching refs:bvm

250                      const BlockAndValueMapping &bvm, LinalgOp linalgOp,  in vectorizeLinalgYield()  argument
258 Value vectorValue = bvm.lookup(outputs.value()); in vectorizeLinalgYield()
307 const BlockAndValueMapping &bvm) { in reduceIfNeeded() argument
308 Value reduceVec = bvm.lookup(reduceValue); in reduceIfNeeded()
309 Value outputVec = bvm.lookup(initialValue); in reduceIfNeeded()
342 const BlockAndValueMapping &bvm, in vectorizeOneOp() argument
349 VectorizationResult result = customFunc(op, bvm); in vectorizeOneOp()
383 reductionOperands[0].second, bvm); in vectorizeOneOp()
392 auto vt = bvm.lookup(operand).getType().dyn_cast<VectorType>(); in vectorizeOneOp()
399 ? bvm.lookup(v) in vectorizeOneOp()
400 : broadcastIfNeeded(b, bvm.lookup(v), firstMaxRankedShape); in vectorizeOneOp()
446 BlockAndValueMapping bvm; in vectorizeAsLinalgGeneric() local
449 bvm.map(valuesSet.getArrayRef(), valuesSet.getArrayRef()); in vectorizeAsLinalgGeneric()
465 bvm.map(bbarg, opOperand->get()); in vectorizeAsLinalgGeneric()
497 bvm.map(bbarg, readValue); in vectorizeAsLinalgGeneric()
498 bvm.map(opOperand->get(), readValue); in vectorizeAsLinalgGeneric()
505 const BlockAndValueMapping &bvm) -> VectorizationResult { in vectorizeAsLinalgGeneric() argument
506 return vectorizeLinalgYield(b, op, bvm, linalgOp, newResults); in vectorizeAsLinalgGeneric()
513 const BlockAndValueMapping &bvm) -> VectorizationResult { in vectorizeAsLinalgGeneric() argument
520 VectorizationResult result = vectorizeOneOp(b, linalgOp, &op, bvm, hooks); in vectorizeAsLinalgGeneric()
527 bvm.map(op.getResults(), result.newOp->getResults()); in vectorizeAsLinalgGeneric()