Searched refs:tensorFromElements (Results 1 – 1 of 1) sorted by relevance
629 LogicalResult matchAndRewrite(GenerateOp tensorFromElements, in matchAndRewrite()632 tensorFromElements.getResult().getType().cast<RankedTensorType>(); in matchAndRewrite()639 auto operandsIt = tensorFromElements.getDynamicExtents().begin(); in matchAndRewrite()656 if (newOperands.size() == tensorFromElements.getDynamicExtents().size()) in matchAndRewrite()659 auto loc = tensorFromElements.getLoc(); in matchAndRewrite()663 rewriter.inlineRegionBefore(tensorFromElements.getBody(), newOp.getBody(), in matchAndRewrite()665 rewriter.replaceOpWithNewOp<tensor::CastOp>(tensorFromElements, resultType, in matchAndRewrite()687 auto tensorFromElements = extract.getTensor().getDefiningOp<GenerateOp>(); in matchAndRewrite() local688 if (!tensorFromElements || !wouldOpBeTriviallyDead(tensorFromElements)) in matchAndRewrite()692 Block *body = &tensorFromElements.getBody().front(); in matchAndRewrite()