Lines Matching defs:dimOp

851   LogicalResult matchAndRewrite(DimOp dimOp,
853     auto castOp = dimOp.getSource().getDefiningOp<CastOp>();
857     rewriter.replaceOpWithNewOp<DimOp>(dimOp, newSource, dimOp.getIndex());
867   LogicalResult matchAndRewrite(DimOp dimOp,
869     auto source = dimOp.getSource();
878         dimOp, [&]() { dimOp.getSourceMutable().assign(initOperand->get()); });
1025   LogicalResult matchAndRewrite(tensor::DimOp dimOp,
1027     std::optional<int64_t> maybeConstantIndex = dimOp.getConstantIndex();
1028     auto emptyTensorOp = dimOp.getSource().getDefiningOp<EmptyOp>();
1036     rewriter.replaceOp(dimOp,
1678       if (auto dimOp = element.getDefiningOp<tensor::DimOp>()) {
1679         dynamicNoop &= dimOp.getSource() == source;
1682         auto cst = getConstantIntValue(dimOp.getIndex());
1967   LogicalResult matchAndRewrite(DimOp dimOp,
1969     auto expandShapeOp = dimOp.getSource().getDefiningOp<ExpandShapeOp>();
1974     std::optional<int64_t> dim = dimOp.getConstantIndex();
1999         rewriter.create<DimOp>(dimOp.getLoc(), expandShapeOp.getSrc(), srcDim);
2001     bindSymbols(dimOp.getContext(), expr);
2003         dimOp, expr.floorDiv(product), srcDimSz);
2011   LogicalResult matchAndRewrite(DimOp dimOp,
2013     auto collapseShapeOp = dimOp.getSource().getDefiningOp<CollapseShapeOp>();
2018     std::optional<int64_t> dim = dimOp.getConstantIndex();
2038           dimOp.getLoc(), collapseShapeOp.getSrc(), it.value()));
2042     rewriter.replaceOpWithNewOp<affine::AffineApplyOp>(dimOp, product,