Searched refs:dilationW (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 1318 int dilationW) in Conv1DNwcGenerator()
1320 dilationW(dilationW) { in Conv1DNwcGenerator()
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()
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()
1649 int strideW, dilationW; member
|