Lines Matching defs:expansionInfo

677                                           const ExpansionInfo &expansionInfo,
681   for (unsigned i : llvm::seq<unsigned>(0, expansionInfo.getOrigOpNumDims())) {
682     ArrayRef<int64_t> expandedShape = expansionInfo.getExpandedShapeOfDim(i);
699                            const ExpansionInfo &expansionInfo) {
704         llvm::map_range(expansionInfo.getExpandedDims(pos), [&](int64_t v) {
709   return AffineMap::get(expansionInfo.getExpandedOpNumDims(),
718                                         const ExpansionInfo &expansionInfo) {
722     auto dimExpansion = expansionInfo.getExpandedShapeOfDim(dim);
736                              const ExpansionInfo &expansionInfo) {
741     auto numExpandedDims = expansionInfo.getExpandedDims(dim).size();
757                                           const ExpansionInfo &expansionInfo) {
762         expansionInfo.getExpandedDims(indexOp.getDim());
774         expansionInfo.getExpandedShapeOfDim(indexOp.getDim()).drop_front();
797                             const ExpansionInfo &expansionInfo,
799   for (unsigned i : llvm::seq<unsigned>(0, expansionInfo.getOrigOpNumDims())) {
800     ArrayRef<int64_t> expandedShape = expansionInfo.getExpandedShapeOfDim(i);
840   ExpansionInfo expansionInfo;
841   if (failed(expansionInfo.compute(
850   if (failed(validateDynamicDimExpansion(linalgOp, expansionInfo, rewriter)))
853   if (failed(isLinalgOpExpandable(linalgOp, expansionInfo, rewriter)))
858         return getIndexingMapInExpandedOp(rewriter, m, expansionInfo);
877           getExpandedType(opOperandType, indexingMap, expansionInfo);
881             getReassociationForExpansion(indexingMap, expansionInfo);
903         getExpandedType(opOperandType, indexingMap, expansionInfo);
906           getReassociationForExpansion(indexingMap, expansionInfo);
924       expansionInfo.getExpandedOpNumDims(), utils::IteratorType::parallel);
926     for (auto j : expansionInfo.getExpandedDims(i))
939   updateExpandedGenericOpRegion(rewriter, loc, fusedRegion, expansionInfo);
951               expansionInfo);