Home
last modified time | relevance | path

Searched refs:wSizeStep (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp1397 int64_t wSizeStep = strideW == 1 ? wSize : 1; in conv() local
1431 for (int64_t w = 0; w < wSize; w += wSizeStep) { in conv()
1445 for (int64_t w = 0; w < wSize; w += wSizeStep) { in conv()
1454 return kw * (wSize / wSizeStep) + w; in conv()
1459 for (int64_t w = 0; w < wSize; w += wSizeStep) { in conv()
1467 for (int64_t w = 0; w < wSize; w += wSizeStep) { in conv()
1521 int64_t wSizeStep = strideW == 1 ? wSize : 1; in depthwiseConv() local
1555 for (int64_t w = 0; w < wSize; w += wSizeStep) { in depthwiseConv()
1569 for (int64_t w = 0; w < wSize; w += wSizeStep) { in depthwiseConv()
1578 return kw * (wSize / wSizeStep) + w; in depthwiseConv()
[all …]