Lines Matching defs:bvm

542     RewriterBase &rewriter, IRMapping &bvm, tensor::PadOp opToHoist,
554   // Step 0. Populate bvm with opToHoist.getSource if relevant.
563     bvm.map(bbArg, operand.get());
574       if (bvm.lookupOrDefault(sliceOp.getSource()) == hoistedPackedTensor) {
584       rewriter.clone(*op, bvm);
591         loc, bvm.lookupOrDefault(forOp.getLowerBound()),
592         bvm.lookupOrDefault(forOp.getUpperBound()),
593         bvm.lookupOrDefault(forOp.getStep()), hoistedPackedTensor);
596     bvm.map(forOp.getInductionVar(), clonedForOp.getInductionVar());
597     bvm.map(forOp.getRegionIterArgs(), clonedForOp.getRegionIterArgs());
598     bvm.map(forOp.getResults(), clonedForOp.getResults());
637   Value paddedTensor = bvm.lookup(opToHoist.getResult());
671       cast<tensor::PadOp>(bvm.lookup(opToHoist.getResult()).getDefiningOp())};
678     RewriterBase &rewriter, IRMapping &bvm, tensor::PadOp opToHoist,
715   return buildPackingLoopNestImpl(rewriter, bvm, opToHoist, transposeVector,
732   IRMapping bvm;
733   return buildPackingLoopNestImpl(rewriter, bvm, opToHoist, transposeVector,
876                                     const IRMapping &bvm,
918     hoistedPackedTensor = bvm.lookup(opToHoist.getResult());
954   IRMapping bvm;
956       rewriter, bvm, opToHoist, transposeVector, analysis);
972       replaceByPackingResult(rewriter, bvm, opToHoist, *transposedTensorType,