Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp536 unsigned origDim = expr.cast<AffineDimExpr>().getPosition(); in fuseWithReshapeByExpansion() local
538 reshapeDims, reshapeDims + numFoldedDims[origDim]); in fuseWithReshapeByExpansion()
540 expandedOpOperandShape.append(expandedDimsShape[origDim].begin(), in fuseWithReshapeByExpansion()
541 expandedDimsShape[origDim].end()); in fuseWithReshapeByExpansion()
542 reshapeDims += numFoldedDims[origDim]; in fuseWithReshapeByExpansion()