Lines Matching refs:SmallVector

160     SmallVector<Value> fusedIndices;  in generateFusedElementwiseOpRegion()
249 static Optional<SmallVector<Value>>
263 SmallVector<Value> fusedOperands; in fuseElementwiseOpsImpl()
264 SmallVector<AffineMap> fusedIndexMaps; in fuseElementwiseOpsImpl()
271 SmallVector<OpOperand *> consumerInputs = consumer.getInputOperands(); in fuseElementwiseOpsImpl()
272 SmallVector<OpOperand *>::iterator it = in fuseElementwiseOpsImpl()
317 SmallVector<Value> consumerOutputs = consumer.getOutputOperands(); in fuseElementwiseOpsImpl()
350 return SmallVector<Value>(fusedOp->getResults()); in fuseElementwiseOpsImpl()
353 static Optional<SmallVector<Value>>
380 Optional<SmallVector<Value>> fusedOpResults = in matchAndRewrite()
501 SmallVector<ReassociationIndices> reassociation;
504 SmallVector<SmallVector<int64_t>> expandedShapeMap;
506 SmallVector<int64_t> originalLoopExtent;
521 SmallVector<int64_t, 4> originalLoopRange = linalgOp.getStaticLoopRanges(); in compute()
528 SmallVector<unsigned> numExpandedDims(fusedIndexMap.getNumDims(), 1); in compute()
587 SmallVector<AffineExpr> newExprs; in getIndexingMapInExpandedOp()
590 SmallVector<AffineExpr, 4> expandedExprs = llvm::to_vector<4>( in getIndexingMapInExpandedOp()
606 SmallVector<int64_t> expandedShape; in getExpandedType()
621 static SmallVector<ReassociationIndices>
624 SmallVector<ReassociationIndices> reassociation; in getReassociationForExpansion()
629 SmallVector<int64_t, 2> indices = llvm::to_vector<2>( in getReassociationForExpansion()
662 SmallVector<Value> expandedIndices; in updateExpandedGenericOpRegion()
683 static Optional<SmallVector<Value>>
711 SmallVector<AffineMap, 4> expandedOpIndexingMaps = llvm::to_vector<4>( in fuseWithReshapeByExpansion()
716 SmallVector<Value> expandedOpOperands; in fuseWithReshapeByExpansion()
731 SmallVector<ReassociationIndices> reassociation = in fuseWithReshapeByExpansion()
751 SmallVector<Value> outputs; in fuseWithReshapeByExpansion()
758 SmallVector<ReassociationIndices> reassociation = in fuseWithReshapeByExpansion()
775 SmallVector<StringRef> iteratorTypes(expansionInfo.getExpandedOpNumDims(), in fuseWithReshapeByExpansion()
792 SmallVector<Value> resultVals; in fuseWithReshapeByExpansion()
796 SmallVector<ReassociationIndices> reassociation = in fuseWithReshapeByExpansion()
840 Optional<SmallVector<Value>> replacementValues = in matchAndRewrite()
875 Optional<SmallVector<Value>> replacementValues = fuseWithReshapeByExpansion( in matchAndRewrite()
1004 static SmallVector<ReassociationIndices>
1018 SmallVector<int64_t> reductionDims; in getCollapsableIterationSpaceDims()
1028 SmallVector<ReassociationIndices> iterationSpaceReassociation; in getCollapsableIterationSpaceDims()
1189 SmallVector<ReassociationIndices> collapsedOpToOrigOpIterationDim;
1193 SmallVector<std::pair<int64_t, unsigned>> origOpToCollapsedOpIterationDim;
1199 static SmallVector<StringRef>
1202 SmallVector<StringRef> collapsedIteratorTypes; in getCollapsedOpIteratorTypes()
1224 SmallVector<AffineExpr> resultExprs; in getCollapsedOpIndexingMap()
1243 static SmallVector<ReassociationIndices>
1247 SmallVector<ReassociationIndices> operandReassociation; in getOperandReassociation()
1276 SmallVector<ReassociationIndices> operandReassociation = in getCollapsedOpOperand()
1332 static FailureOr<SmallVector<Value>> collapseGenericOpIterationDims( in collapseGenericOpIterationDims()
1350 SmallVector<StringRef> iteratorTypes = getCollapsedOpIteratorTypes( in collapseGenericOpIterationDims()
1369 SmallVector<Type> resultTypes; in collapseGenericOpIterationDims()
1370 SmallVector<Value> outputOperands; in collapseGenericOpIterationDims()
1393 SmallVector<Range> loopRanges = in collapseGenericOpIterationDims()
1402 SmallVector<Value> loopBound = llvm::to_vector( in collapseGenericOpIterationDims()
1411 SmallVector<Value> results; in collapseGenericOpIterationDims()
1421 SmallVector<ReassociationIndices> reassociation = in collapseGenericOpIterationDims()
1454 SmallVector<ReassociationIndices> collapsableIterationDims = in matchAndRewrite()
1462 Optional<SmallVector<Value>> replacements = in matchAndRewrite()
1534 SmallVector<AffineMap> fusedIndexMaps; in matchAndRewrite()
1535 SmallVector<Value> fusedOperands; in matchAndRewrite()
1536 SmallVector<Location> fusedLocs{genericOp.getLoc()}; in matchAndRewrite()
1561 SmallVector<Value> outputOperands = genericOp.getOutputOperands(); in matchAndRewrite()
1623 SmallVector<Value> dynamicDims; in matchAndRewrite()