Searched refs:dimOp (Results 1 – 3 of 3) sorted by relevance
292 auto dimOp = val.value().getDefiningOp<memref::DimOp>(); in tryEvaluateShapeToRoot() local293 if (!dimOp) return false; in tryEvaluateShapeToRoot()294 if (!candidate) candidate = dimOp.getSource(); in tryEvaluateShapeToRoot()295 auto index = dimOp.getConstantIndex(); in tryEvaluateShapeToRoot()297 return candidate == dimOp.getSource() && in tryEvaluateShapeToRoot()
828 LogicalResult matchAndRewrite(OpTy dimOp, in matchAndRewrite()830 auto src = dimOp.getSource().template dyn_cast<BlockArgument>(); in matchAndRewrite()844 rewriter.updateRootInPlace(dimOp, [&] { in matchAndRewrite()845 dimOp.getSourceMutable().assign( in matchAndRewrite()854 rewriter.updateRootInPlace(dimOp, [&] { in matchAndRewrite()855 dimOp.getSourceMutable().assign( in matchAndRewrite()888 LogicalResult matchAndRewrite(OpTy dimOp, in matchAndRewrite()890 auto loopOp = dimOp.getSource().template getDefiningOp<LoopOp>(); in matchAndRewrite()892 auto opResult = dimOp.getSource().template cast<OpResult>(); in matchAndRewrite()895 rewriter.updateRootInPlace(dimOp, [&]() { in matchAndRewrite()[all …]
155 Value dimOp = rewriter.create<tensor::DimOp>(loc, rewriter.getIndexType(), in matchAndRewrite() local159 Value cast = rewriter.create<arith::IndexCastOp>(loc, elementTy, dimOp); in matchAndRewrite()