Searched refs:fusedIndexMaps (Results 1 – 1 of 1) sorted by relevance
202 SmallVector<Attribute, 4> fusedIndexMaps; in fuseTensorOpsImpl() local204 fusedIndexMaps.reserve(fusedOperands.size() + consumer.getNumOutputs()); in fuseTensorOpsImpl()205 fusedIndexMaps.assign(consumerIndexMaps.begin(), in fuseTensorOpsImpl()209 producer, consumer.getInputIndexingMap(consumerIdx), fusedIndexMaps); in fuseTensorOpsImpl()212 fusedIndexMaps.append(std::next(consumerIndexMaps.begin(), consumerIdx + 1), in fuseTensorOpsImpl()226 rewriter.getArrayAttr(fusedIndexMaps), in fuseTensorOpsImpl()239 rewriter.getArrayAttr(fusedIndexMaps), in fuseTensorOpsImpl()710 SmallVector<AffineMap, 4> fusedIndexMaps = llvm::to_vector<4>( in matchAndRewrite() local714 auto invMap = inversePermutation(fusedIndexMaps[operand.index()]); in matchAndRewrite()724 fusedIndexMaps[operand.index()] = modifiedMap; in matchAndRewrite()[all …]