Searched defs:flatten1DDepthwiseConv (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 1744 vectorizeDynamicConvOpPrecondition(linalg::LinalgOp conv,bool flatten1DDepthwiseConv) vectorizeDynamicConvOpPrecondition() argument
1772 vectorizeDynamicLinalgOpPrecondition(linalg::LinalgOp op,bool flatten1DDepthwiseConv) vectorizeDynamicLinalgOpPrecondition() argument
1811 vectorizeLinalgOpPrecondition(LinalgOp linalgOp,ArrayRef<int64_t> inputVectorSizes,bool vectorizeNDExtract,bool flatten1DDepthwiseConv) vectorizeLinalgOpPrecondition() argument
1959 vectorizeOpPrecondition(Operation * op,ArrayRef<int64_t> inputVectorSizes,ArrayRef<bool> inputScalableVecDims,bool vectorizeNDExtract,bool flatten1DDepthwiseConv) vectorizeOpPrecondition() argument
2007 vectorize(RewriterBase & rewriter,Operation * op,ArrayRef<int64_t> inputVectorSizes,ArrayRef<bool> inputScalableVecDims,bool vectorizeNDExtract,bool flatten1DDepthwiseConv) vectorize() argument
3586 vectorizeConvolution(RewriterBase & rewriter,LinalgOp op,ArrayRef<int64_t> inputVecSizes,ArrayRef<bool> inputScalableVecDims,bool flatten1DDepthwiseConv) vectorizeConvolution() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 3271 build(OpBuilder & builder,OperationState & result,Value target,bool vectorizePadding,bool vectorizeExtract,bool flatten1DDepthwiseConv) build() argument
3321 bool flatten1DDepthwiseConv = false; global() member
|