Lines Matching refs:SmallVector
108 SmallVector<AffineExpr, 4> dimReplacements; in replaceUnitDims()
119 SmallVector<AffineExpr, 4> symReplacements; in replaceUnitDims()
124 SmallVector<AffineMap, 4> newIndexingMaps; in replaceUnitDims()
174 SmallVector<AffineMap, 4> indexingMaps = genericOp.getIndexingMapsArray(); in matchAndRewrite()
184 SmallVector<int64_t> dims = genericOp.getStaticShape(); in matchAndRewrite()
187 SmallVector<unsigned, 4> unitDimsReductionLoops; in matchAndRewrite()
207 SmallVector<Attribute, 4> newIteratorTypes; in matchAndRewrite()
245 SmallVector<AffineExpr> reassociations; in replaceUnitExtents()
246 SmallVector<Attribute> reassociationMaps; in replaceUnitExtents()
247 SmallVector<AffineExpr> newIndexExprs; in replaceUnitExtents()
248 SmallVector<int64_t> newShape; in replaceUnitExtents()
304 SmallVector<ReassociationExprs, 2>
306 SmallVector<ReassociationExprs, 2> reassociationExprs; in convertAffineMapArrayToExprs()
369 SmallVector<AffineMap> newIndexingMaps; in matchAndRewrite()
370 SmallVector<ArrayAttr> reassociationMaps; in matchAndRewrite()
371 SmallVector<Type> newInputOutputTypes; in matchAndRewrite()
409 SmallVector<Value, 4> res; in matchAndRewrite()
423 SmallVector<Value, 4> newInputs = insertReshapes(genericOp.inputs()); in matchAndRewrite()
424 SmallVector<Value, 4> newOutputs = insertReshapes(genericOp.outputs()); in matchAndRewrite()
428 SmallVector<Type, 4> resultTypes; in matchAndRewrite()
441 SmallVector<Value, 4> resultReplacements; in matchAndRewrite()
467 SmallVector<OpFoldResult> offsets = sliceOp.getMixedOffsets(); in matchAndRewrite()
468 SmallVector<OpFoldResult> sizes = sliceOp.getMixedSizes(); in matchAndRewrite()
469 SmallVector<OpFoldResult> strides = sliceOp.getMixedStrides(); in matchAndRewrite()
498 SmallVector<OpFoldResult> offsets = insertSliceOp.getMixedOffsets(); in matchAndRewrite()
499 SmallVector<OpFoldResult> sizes = insertSliceOp.getMixedSizes(); in matchAndRewrite()
500 SmallVector<OpFoldResult> strides = insertSliceOp.getMixedStrides(); in matchAndRewrite()