Lines Matching refs:InitTensorOp
544 auto newInitOp = rewriter.create<InitTensorOp>( in matchAndRewrite()
1189 void InitTensorOp::build(OpBuilder &b, OperationState &result, in build()
1201 LogicalResult InitTensorOp::verify() { in verify()
1215 Type expectedType = InitTensorOp::inferResultType( in verify()
1225 Type InitTensorOp::inferResultType(ArrayRef<int64_t> staticSizes, in inferResultType()
1230 SmallVector<OpFoldResult> InitTensorOp::getMixedSizes() { in getMixedSizes()
1257 struct ReplaceStaticShapeDims : OpRewritePattern<InitTensorOp> {
1258 using OpRewritePattern<InitTensorOp>::OpRewritePattern;
1260 LogicalResult matchAndRewrite(InitTensorOp op, in matchAndRewrite()
1290 rewriter.create<InitTensorOp>(op.getLoc(), newType, dynamicSizes, in matchAndRewrite()
1309 if (!sliceOp.getSource().getDefiningOp<linalg::InitTensorOp>()) in matchAndRewrite()
1313 rewriter.replaceOpWithNewOp<linalg::InitTensorOp>( in matchAndRewrite()
1328 if (!reshapeOp.getSrc().template getDefiningOp<InitTensorOp>()) in matchAndRewrite()
1338 Value initTensor = rewriter.create<InitTensorOp>( in matchAndRewrite()
1357 auto initTensorOp = dimOp.getSource().getDefiningOp<linalg::InitTensorOp>(); in matchAndRewrite()
1392 auto producer = castOp.getSource().getDefiningOp<InitTensorOp>(); in matchAndRewrite()
1434 rewriter.replaceOpWithNewOp<InitTensorOp>(castOp, newMixedSizes, in matchAndRewrite()
1442 void InitTensorOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns()
1451 LogicalResult InitTensorOp::reifyResultShapes( in reifyResultShapes()