Searched refs:tiledOperands (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | BubbleUpExtractSlice.cpp | 122 SmallVector<Value, 4> tiledOperands = makeTiledShapes( in matchAndRewrite() local
129 tiledOperands[opOperand->getOperandNumber()].getType()); in matchAndRewrite()
132 linalgOp.clone(rewriter, linalgLoc, resultTensorTypes, tiledOperands); in matchAndRewrite()
|
| H A D | FusionOnTensors.cpp | 165 SmallVector<Value> tiledOperands = producerOp.getInputAndOutputOperands(); in getTiledProducer() local
166 tiledOperands = makeTiledShapes(b, loc, producerOp, tiledOperands, tileIvs, in getTiledProducer()
179 tiledOperands[outputOperand->getOperandNumber()] = sliceOp.getResult(); in getTiledProducer()
183 TypeRange resultTypes = ValueRange(tiledOperands) in getTiledProducer()
186 LinalgOp clonedOp = producerOp.clone(b, loc, resultTypes, tiledOperands); in getTiledProducer()
|
| H A D | TilingInterfaceImpl.cpp | 76 SmallVector<Value, 4> tiledOperands = makeTiledShapes( in getTiledImplementation() local
82 return tiledOperands[opOperand->getOperandNumber()].getType(); in getTiledImplementation()
86 linalgOp.clone(b, loc, resultTensorTypes, tiledOperands); in getTiledImplementation()
|
| H A D | Tiling.cpp | 466 SmallVector<Value, 4> tiledOperands = in tileLinalgOpImpl() local
471 getTensorOutputTypes(op, tiledOperands); in tileLinalgOpImpl()
472 res = op.clone(b, loc, resultTensorTypes, tiledOperands); in tileLinalgOpImpl()
474 insertSlicesBack(builder, loc, op, tiledOperands, res->getResults()); in tileLinalgOpImpl()
|