Searched refs:tensorFromElements (Results 1 – 1 of 1) sorted by relevance
1809 LogicalResult matchAndRewrite(DynamicTensorFromElementsOp tensorFromElements, in matchAndRewrite()1812 tensorFromElements.getResult().getType().cast<RankedTensorType>(); in matchAndRewrite()1819 auto operandsIt = tensorFromElements.dynamicExtents().begin(); in matchAndRewrite()1836 if (newOperands.size() == tensorFromElements.dynamicExtents().size()) in matchAndRewrite()1839 auto loc = tensorFromElements.getLoc(); in matchAndRewrite()1843 rewriter.inlineRegionBefore(tensorFromElements.body(), newOp.body(), in matchAndRewrite()1845 rewriter.replaceOpWithNewOp<TensorCastOp>(tensorFromElements, resultType, in matchAndRewrite()1868 auto tensorFromElements = in matchAndRewrite() local1870 if (!tensorFromElements || !wouldOpBeTriviallyDead(tensorFromElements)) in matchAndRewrite()1874 Block *body = tensorFromElements.getBody(); in matchAndRewrite()[all …]