Searched refs:rank_type (Results 1 – 2 of 2) sorted by relevance
686 RankedTensorType rank_type = in matchAndRewrite() local688 auto rank_attr = DenseElementsAttr::get(rank_type, {rank}); in matchAndRewrite()690 rewriter.create<tosa::ConstOp>(op->getLoc(), rank_type, rank_attr); in matchAndRewrite()
1604 RankedTensorType rank_type = in matchAndRewrite() local1606 auto rank_attr = DenseElementsAttr::get(rank_type, {rank}); in matchAndRewrite()1608 rewriter.create<tosa::ConstOp>(op->getLoc(), rank_type, rank_attr); in matchAndRewrite()