Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp503 for (auto numFoldedDim : llvm::enumerate(numFoldedDims)) { in fuseWithReshapeByExpansion() local
504 sum += numFoldedDim.value(); in fuseWithReshapeByExpansion()
505 remapping[numFoldedDim.index() + 1] = sum; in fuseWithReshapeByExpansion()