Lines Matching refs:getDefiningOp
44 if (auto inputOp = input.getDefiningOp<ShapeOfOp>()) { in getShapeVec()
50 } else if (auto inputOp = input.getDefiningOp<ConstShapeOp>()) { in getShapeVec()
53 } else if (auto inputOp = input.getDefiningOp<ConstantOp>()) { in getShapeVec()
290 auto witness = op.witness().getDefiningOp<ConstWitnessOp>(); in matchAndRewrite()
420 auto cstrEqOp = w.getDefiningOp<CstrEqOp>(); in matchAndRewrite()
549 if (auto constShape = shape.getDefiningOp<ConstShapeOp>()) { in matchAndRewrite()
607 if (auto constShape = shape.getDefiningOp<ConstShapeOp>()) { in matchAndRewrite()
646 if (auto castOp = operand.getDefiningOp<tensor::CastOp>()) { in matchAndRewrite()
1066 if (auto constSizeOp = dim().getDefiningOp<ConstSizeOp>()) in getConstantDim()
1068 if (auto constantOp = dim().getDefiningOp<ConstantOp>()) in getConstantDim()
1151 auto shapeOfOp = op.shape().getDefiningOp<ShapeOfOp>(); in matchAndRewrite()
1300 auto shapeOfOp = op.source().getDefiningOp<ShapeOfOp>(); in matchAndRewrite()