Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp466 AffineMap foldedDims = reassociationMaps[resultExpr.index()]; in fuseWithReshapeByExpansion() local
467 numFoldedDims[pos] = foldedDims.getNumResults(); in fuseWithReshapeByExpansion()
469 expandedShape.slice(foldedDims.getDimPosition(0), numFoldedDims[pos]); in fuseWithReshapeByExpansion()
537 auto foldedDims = llvm::seq<int64_t>( in fuseWithReshapeByExpansion() local
539 reassociation.emplace_back(foldedDims.begin(), foldedDims.end()); in fuseWithReshapeByExpansion()