Lines Matching refs:operation
30 matchAndReplaceDepthwiseConv(Operation *operation, Value input, Value kernel, in matchAndReplaceDepthwiseConv() argument
33 Location loc = operation->getLoc(); in matchAndReplaceDepthwiseConv()
34 auto linalgOp = dyn_cast<LinalgOp>(operation); in matchAndReplaceDepthwiseConv()
39 auto result = operation->getResult(0); in matchAndReplaceDepthwiseConv()
71 if (isa<DepthwiseConv2DNhwcHwcmOp>(operation)) { in matchAndReplaceDepthwiseConv()
77 } else if (isa<DepthwiseConv2DNhwcHwcmQOp>(operation)) { in matchAndReplaceDepthwiseConv()
91 operation, resultTy, newConv, collapsedInitDims); in matchAndReplaceDepthwiseConv()
102 Operation *operation = op.getOperation(); in matchAndRewrite() local
110 return matchAndReplaceDepthwiseConv(operation, input, kernel, nullptr, in matchAndRewrite()
122 Operation *operation = op.getOperation(); in matchAndRewrite() local
132 return matchAndReplaceDepthwiseConv(operation, input, kernel, iZp, kZp, in matchAndRewrite()