Home
last modified time | relevance | path

Searched refs:extractSliceOp (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DHoisting.cpp54 tensor::ExtractSliceOp extractSliceOp; member
174 user == candidateRead.extractSliceOp || in tensorChunkAccessedByUnknownOp()
266 assert(((read.extractSliceOp && write.insertSliceOp) || in hoistReadWrite()
267 (!read.extractSliceOp && !write.insertSliceOp)) && in hoistReadWrite()
276 if (read.extractSliceOp) in hoistReadWrite()
277 forOp.moveOutOfLoop(read.extractSliceOp); in hoistReadWrite()
287 if (read.extractSliceOp) in hoistReadWrite()
288 read.extractSliceOp.getSourceMutable().assign( in hoistReadWrite()
324 read.extractSliceOp.getResult()); in hoistReadWrite()
326 read.extractSliceOp.getSource()); in hoistReadWrite()
H A DFusionOnTensors.cpp253 if (auto extractSliceOp = dyn_cast<tensor::ExtractSliceOp>(op)) { in hasOtherUses() local
254 if (extractSliceOp != sliceOp) in hasOtherUses()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp280 auto extractSliceOp = cast<tensor::ExtractSliceOp>(op); in bufferize() local
282 SmallVector<OpFoldResult> mixedSizes = extractSliceOp.getMixedSizes(); in bufferize()
284 Location loc = extractSliceOp.getLoc(); in bufferize()
288 getBuffer(rewriter, extractSliceOp.getSource(), options); in bufferize()
296 extractSliceOp.getType().getShape(), srcMemrefType, mixedOffsets, in bufferize()
557 ExtractSliceOp extractSliceOp, in areEquivalentExtractSliceOps() argument
559 if (!extractSliceOp || !insertSliceOp) in areEquivalentExtractSliceOps()
561 if (extractSliceOp != insertSliceOp && in areEquivalentExtractSliceOps()
562 !state.areEquivalentBufferizedValues(extractSliceOp.getSource(), in areEquivalentExtractSliceOps()
565 if (!sameOffsetsSizesAndStrides(extractSliceOp, insertSliceOp, in areEquivalentExtractSliceOps()
[all …]