Lines Matching defs:tosa

23 namespace tosa {
26 } // namespace tosa
30 using namespace mlir::tosa;
108 struct ConvertTosaOp<tosa::MulOp> : public OpRewritePattern<tosa::MulOp> {
109   using OpRewritePattern<tosa::MulOp>::OpRewritePattern;
111   LogicalResult matchAndRewrite(tosa::MulOp tosaBinaryOp,
127     rewriter.replaceOpWithNewOp<tosa::MulOp>(tosaBinaryOp, outputType, input1,
138 struct ConvertTosaOp<tosa::ArithmeticRightShiftOp>
139     : public OpRewritePattern<tosa::ArithmeticRightShiftOp> {
140   using OpRewritePattern<tosa::ArithmeticRightShiftOp>::OpRewritePattern;
142   LogicalResult matchAndRewrite(tosa::ArithmeticRightShiftOp tosaBinaryOp,
158     rewriter.replaceOpWithNewOp<tosa::ArithmeticRightShiftOp>(
166 struct ConvertTosaOp<tosa::SelectOp> : public OpRewritePattern<tosa::SelectOp> {
167   using OpRewritePattern<tosa::SelectOp>::OpRewritePattern;
169   LogicalResult matchAndRewrite(tosa::SelectOp tosaOp,
210     rewriter.replaceOpWithNewOp<tosa::SelectOp>(tosaOp, outputType, input1,
222     : public tosa::impl::TosaMakeBroadcastableBase<TosaMakeBroadcastable> {
229     patterns.add<ConvertTosaOp<tosa::BitwiseAndOp>>(ctx);
230     patterns.add<ConvertTosaOp<tosa::BitwiseOrOp>>(ctx);
231     patterns.add<ConvertTosaOp<tosa::BitwiseXorOp>>(ctx);
232     patterns.add<ConvertTosaOp<tosa::AddOp>>(ctx);
233     patterns.add<ConvertTosaOp<tosa::SubOp>>(ctx);
234     patterns.add<ConvertTosaOp<tosa::MulOp>>(ctx);
235     patterns.add<ConvertTosaOp<tosa::IntDivOp>>(ctx);
236     patterns.add<ConvertTosaOp<tosa::MaximumOp>>(ctx);
237     patterns.add<ConvertTosaOp<tosa::MinimumOp>>(ctx);
238     patterns.add<ConvertTosaOp<tosa::EqualOp>>(ctx);
239     patterns.add<ConvertTosaOp<tosa::GreaterOp>>(ctx);
240     patterns.add<ConvertTosaOp<tosa::GreaterEqualOp>>(ctx);
241     patterns.add<ConvertTosaOp<tosa::LogicalLeftShiftOp>>(ctx);
242     patterns.add<ConvertTosaOp<tosa::ArithmeticRightShiftOp>>(ctx);
243     patterns.add<ConvertTosaOp<tosa::LogicalRightShiftOp>>(ctx);
244     patterns.add<ConvertTosaOp<tosa::LogicalAndOp>>(ctx);
245     patterns.add<ConvertTosaOp<tosa::LogicalOrOp>>(ctx);
246     patterns.add<ConvertTosaOp<tosa::LogicalXorOp>>(ctx);
247     patterns.add<ConvertTosaOp<tosa::SelectOp>>(ctx);
248     patterns.add<ConvertTosaOp<tosa::PowOp>>(ctx);
254 std::unique_ptr<Pass> mlir::tosa::createTosaMakeBroadcastablePass() {