Searched refs:shapeOfOp (Results 1 – 2 of 2) sorted by relevance
1371 auto shapeOfOp = op.getShape().getDefiningOp<ShapeOfOp>(); in matchAndRewrite() local1372 if (!shapeOfOp) in matchAndRewrite()1375 shapeOfOp.getArg().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1600 auto shapeOfOp = op.getSource().getDefiningOp<ShapeOfOp>(); in matchAndRewrite() local1601 if (!shapeOfOp) in matchAndRewrite()1605 auto argTy = shapeOfOp.getArg().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1609 rewriter.replaceOpWithNewOp<ShapeOfOp>(op, ty, shapeOfOp.getArg()); in matchAndRewrite()
344 if (auto shapeOfOp = op.getShape().getDefiningOp<ShapeOfOp>()) { in matchAndRewrite() local345 if (shapeOfOp.getArg().getType().isa<ShapedType>()) { in matchAndRewrite()346 rewriter.replaceOpWithNewOp<tensor::DimOp>(op, shapeOfOp.getArg(), in matchAndRewrite()