Searched refs:allocTensorOp (Results 1 – 3 of 3) sorted by relevance
47 Operation *allocTensorOp) { in insertionPointDominatesUses() argument48 for (Operation *user : allocTensorOp->getUsers()) in insertionPointDominatesUses()57 findValidInsertionPoint(Operation *allocTensorOp, in findValidInsertionPoint() argument64 insertionPointCandidates.push_back(allocTensorOp); in findValidInsertionPoint()87 if (!insertionPointDominatesUses(domInfo, insertionPoint, allocTensorOp)) in findValidInsertionPoint()
106 auto allocTensorOp = b.create<AllocTensorOp>(loc, tensorType, dynamicSizes, in allocateTensorForShapedValue() local108 allocTensorOp->setAttr(BufferizationDialect::kEscapeAttrName, in allocateTensorForShapedValue()113 return allocTensorOp.getResult(); in allocateTensorForShapedValue()117 allocTensorOp.setMemorySpaceAttr( in allocateTensorForShapedValue()120 return allocTensorOp.getResult(); in allocateTensorForShapedValue()
335 auto allocTensorOp = dimOp.getSource().getDefiningOp<AllocTensorOp>(); in matchAndRewrite() local336 if (!allocTensorOp || !maybeConstantIndex) in matchAndRewrite()338 if (!allocTensorOp.getType().isDynamicDim(*maybeConstantIndex)) in matchAndRewrite()341 dimOp, allocTensorOp.getDynamicSize(rewriter, *maybeConstantIndex)); in matchAndRewrite()