Searched refs:dstStrides (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/ExecutionEngine/ |
| H A D | CRunnerUtils.cpp | 72 int64_t *dstStrides = static_cast<int64_t *>(alloca(sizeof(int64_t) * rank)); in memrefCopy() local
78 dstStrides[rankp] = dst.strides[rankp] * elemSize; in memrefCopy()
90 writeIndex += dstStrides[axis]; in memrefCopy()
101 writeIndex -= dst.sizes[axis] * dstStrides[axis]; in memrefCopy()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorUnrollDistribute.cpp | 396 SmallVector<int64_t> dstStrides(it.first.size(), 1); in matchAndRewrite() local
398 loc, it.second, result, it.first, dstStrides); in matchAndRewrite()
470 SmallVector<int64_t> dstStrides(it.first.size(), 1); in matchAndRewrite() local
472 loc, it.second, result, it.first, dstStrides); in matchAndRewrite()
|