Searched refs:strideW (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 1317 Conv1DNwcGenerator(OpBuilder &builder, LinalgOp linalgOp, int strideW, in Conv1DNwcGenerator()
1319 : StructuredGenerator<LinalgOp>(builder, linalgOp), strideW(strideW), in Conv1DNwcGenerator()
1397 int64_t wSizeStep = strideW == 1 ? wSize : 1; in conv()
1407 ((wSize - 1) * strideW + 1) + ((kwSize - 1) * dilationW + 1) - 1, in conv()
1434 /*offsets=*/ArrayRef<int64_t>{0, w * strideW + kw * dilationW, 0}, in conv()
1521 int64_t wSizeStep = strideW == 1 ? wSize : 1; in depthwiseConv()
1530 ((wSize - 1) * strideW + 1) + ((kwSize - 1) * dilationW + 1) - 1, in depthwiseConv()
1558 /*offsets=*/ArrayRef<int64_t>{0, w * strideW + kw * dilationW, 0}, in depthwiseConv()
1624 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c}, in generateConv()
1640 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c}, in generateDilatedConv()
[all …]
|