Lines Matching refs:tensorFromElements
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() local
1870 if (!tensorFromElements || !wouldOpBeTriviallyDead(tensorFromElements)) in matchAndRewrite()
1874 Block *body = tensorFromElements.getBody(); in matchAndRewrite()
1996 auto tensorFromElements = dyn_cast_or_null<TensorFromElementsOp>( in matchAndRewrite() local
1998 if (tensorFromElements == nullptr) in matchAndRewrite()
2005 tensorFromElements.getOperand(index.getZExtValue())); in matchAndRewrite()