Home
last modified time | relevance | path

Searched refs:initTensor (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusePadOpWithLinalgProducer.cpp79 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
86 rewriter.create<linalg::FillOp>(loc, padValue, initTensor.getResult()); in matchAndRewrite()
H A DDecomposeLinalgOps.cpp188 Value initTensor = rewriter.create<linalg::InitTensorOp>( in createPeeledGenericOp() local
190 newInitValues.push_back(initTensor); in createPeeledGenericOp()
191 newResultTypes.push_back(initTensor.getType()); in createPeeledGenericOp()
H A DHoistPadding.cpp546 Value initTensor = in hoistPaddingOnTensors() local
550 makeTransposeOp(b, loc, newResult, initTensor, transposeVector)); in hoistPaddingOnTensors()
H A DTransforms.cpp700 Value initTensor = rewriter.create<InitTensorOp>( in matchAndRewrite() local
706 ValueRange{initTensor}) in matchAndRewrite()
H A DElementwiseOpFusion.cpp1630 Value initTensor = rewriter.create<InitTensorOp>( in matchAndRewrite() local
1633 op->setOperand(opOperand->getOperandNumber(), initTensor); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp244 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
249 ValueRange{initTensor}) in matchAndRewrite()
429 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
434 ValueRange{initTensor}) in matchAndRewrite()
530 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
534 ValueRange{initTensor}) in matchAndRewrite()
600 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
608 ValueRange{initTensor}) in matchAndRewrite()
735 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
741 ValueRange{initTensor}) in matchAndRewrite()
H A DTosaToLinalg.cpp810 auto initTensor = rewriter in reduceMatchAndRewriteHelper() local
823 ValueRange{initTensor}) in reduceMatchAndRewriteHelper()
1130 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
1244 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
1366 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
1731 auto initTensor = rewriter in matchAndRewrite() local
1802 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite() local
1820 ValueRange{initTensor}, affineMaps, in matchAndRewrite()
2062 auto initTensor = in matchAndRewrite() local
2077 ValueRange{initTensor}, affineMaps, in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp32 Value initTensor = b.create<linalg::InitTensorOp>( in getDestinationOperands() local
34 return {initTensor}; in getDestinationOperands()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1338 Value initTensor = rewriter.create<InitTensorOp>( in matchAndRewrite() local
1341 if (initTensor.getType() != reshapeOp.getResultType()) { in matchAndRewrite()
1343 reshapeOp, reshapeOp.getResultType(), initTensor); in matchAndRewrite()
1345 rewriter.replaceOp(reshapeOp, initTensor); in matchAndRewrite()