Lines Matching defs:controlFn

375                                const ControlPropagationFn &controlFn) {
381   if (!controlFn(&packOp.getSourceMutable()))
460       : OpRewritePattern<tensor::PackOp>(context), controlFn(std::move(fun)) {}
465         bubbleUpPackOpThroughGenericOp(rewriter, packOp, controlFn);
473   ControlPropagationFn controlFn;
482       : OpRewritePattern<tensor::PackOp>(context), controlFn(std::move(fun)) {}
491     if (!controlFn(&packOp.getSourceMutable()))
566   ControlPropagationFn controlFn;
839       : OpRewritePattern<tensor::PackOp>(context), controlFn(std::move(fun)) {}
857     if (!controlFn(&packOp.getSourceMutable()))
871   ControlPropagationFn controlFn;
895     PatternRewriter &rewriter, ControlPropagationFn controlFn) {
897   if (!controlFn(&expandOp.getSrcMutable()))
968       : OpRewritePattern<tensor::UnPackOp>(context), controlFn(std::move(fun)) {
989                                                     controlFn);
995   ControlPropagationFn controlFn;
1052                                  ControlPropagationFn controlFn) {
1070   if (!controlFn(unPackedOperand))
1124       : OpRewritePattern<GenericOp>(context), controlFn(std::move(fun)) {}
1129         pushDownUnPackOpThroughGenericOp(rewriter, genericOp, controlFn);
1137   ControlPropagationFn controlFn;
1145       : OpRewritePattern<tensor::PadOp>(context), controlFn(std::move(fun)) {}
1154     if (!controlFn(&padOp.getSourceMutable()))
1201   ControlPropagationFn controlFn;