Searched refs:collapsedKernel (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | NamedOpConversions.cpp | 56 auto collapsedKernel = rewriter.create<tensor::CollapseShapeOp>( in matchAndReplaceDepthwiseConv() local
74 loc, newInitTy, ValueRange{input, collapsedKernel}, in matchAndReplaceDepthwiseConv()
81 loc, newInitTy, ValueRange{input, collapsedKernel, iZp, kZp}, in matchAndReplaceDepthwiseConv()
|