Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp375 Value input, int64_t firstDimToCollapse) { in collapseInnerDims() argument
380 for (int64_t i = 0; i < firstDimToCollapse; ++i) in collapseInnerDims()
383 for (int64_t i = firstDimToCollapse; i < inputType.getRank(); ++i) in collapseInnerDims()
393 checkAndCollapseInnerZeroIndices(ValueRange indices, int64_t firstDimToCollapse, in checkAndCollapseInnerZeroIndices() argument
396 if (firstDimToCollapse >= rank) in checkAndCollapseInnerZeroIndices()
398 for (int64_t i = firstDimToCollapse; i < rank; ++i) { in checkAndCollapseInnerZeroIndices()
405 outIndices.resize(firstDimToCollapse + 1); in checkAndCollapseInnerZeroIndices()