Lines Matching refs:sliceOp
130 tensor::ExtractSliceOp sliceOp, in getTiledProducer() argument
137 b.setInsertionPointAfter(sliceOp); in getTiledProducer()
148 SmallVector<Range> sliceOpRanges = sliceOp.getOrCreateRanges(b, loc); in getTiledProducer()
179 tiledOperands[outputOperand->getOperandNumber()] = sliceOp.getResult(); in getTiledProducer()
246 tensor::ExtractSliceOp sliceOp) { in hasOtherUses() argument
254 if (extractSliceOp != sliceOp) in hasOtherUses()
263 if (backwardSlice.empty() || backwardSlice.front() != sliceOp) in hasOtherUses()
342 auto sliceOp = in fuseProducer() local
344 if (!sliceOp) in fuseProducer()
349 if (sliceOp->getBlock() != rootOp->getBlock() || in fuseProducer()
361 auto producerResult = sliceOp.getSource().dyn_cast<OpResult>(); in fuseProducer()
362 if (auto bbArg = sliceOp.getSource().dyn_cast<BlockArgument>()) { in fuseProducer()
365 if (!iterArg || hasOtherUses(bbArg, sliceOp)) in fuseProducer()
384 getTiledProducer(b, producerResult, sliceOp, tiledSliceDimIndices, in fuseProducer()
399 sliceOp.getResult().replaceAllUsesExcept(newResult, clonedOp); in fuseProducer()