Lines Matching defs:conv

1808 vectorizeDynamicConvOpPrecondition(linalg::LinalgOp conv,
1816   if (!isa<linalg::DepthwiseConv1DNwcWcOp>(conv)) {
1817     LDBG("Not a 1D depth-wise WC conv, dynamic shapes are not supported\n");
1823   Value lhs = conv.getDpsInputOperand(0)->get();
3125   FailureOr<Operation *> conv(Conv1DOpOrder conv1DOpOrder) {
3127       return rewriter.notifyMatchFailure(op, "unvectorizable 1-D conv/pool");
3412       return rewriter.notifyMatchFailure(op, "unvectorizable depthwise conv");
3430            "Unsupported flattened conv with dynamic shapes");
3656                                          "failed to match conv::W 1-par 1-red");
3662       return conv(Conv1DOpOrder::W);
3664     return rewriter.notifyMatchFailure(op, "not a conv::W layout");
3674           op, "failed to match conv::Nwc 3-par 2-red");
3680       return conv(Conv1DOpOrder::Nwc);
3682     return rewriter.notifyMatchFailure(op, "not a conv::Nwc layout");
3692           op, "failed to match conv::Ncw 3-par 2-red");
3697       return conv(Conv1DOpOrder::Ncw);
3699     return rewriter.notifyMatchFailure(op, "not a conv::Ncw layout");
3715       return conv(Conv1DOpOrder::Nwc);
3732       return conv(Conv1DOpOrder::Ncw);
3746           op, "failed to match depthwise::Nwc conv 3-par 1-red");
3769   // Sets oper, poolExtOp and isPoolExt for valid conv/pooling ops.
3770   // Returns true iff it is a valid conv/pooling op.
3773   // If conv, check for single `mul` predecessor. The `mul` operands must be
3827   // conv. matcher in the ConvGenerator succeed or fail.
3859            "Not a 1D depthwise conv!");
3862             .Case<linalg::DepthwiseConv1DNwcWcOp>([](auto conv) { return 2; })
3863             .Case<linalg::DepthwiseConv1DNcwCwOp>([](auto conv) { return 1; });