Lines Matching refs:SmallVector
93 SmallVector<unsigned> dims; in splitReduction()
97 SmallVector<int64_t, 4> loopRanges = op.getStaticLoopRanges(); in splitReduction()
105 SmallVector<Operation *, 4> combinerOps; in splitReduction()
116 SmallVector<Value> newInputs; in splitReduction()
117 SmallVector<AffineMap> newMaps; in splitReduction()
121 SmallVector<int64_t> newShape; in splitReduction()
122 SmallVector<AffineExpr> exprs; in splitReduction()
123 SmallVector<ReassociationIndices> reassociation; in splitReduction()
158 SmallVector<int64_t> newOutputShape; in splitReduction()
161 SmallVector<AffineExpr> outputExpr; in splitReduction()
193 SmallVector<StringRef> newIteratorTypes; in splitReduction()
211 SmallVector<Value> outputOperands = op.getOutputOperands(); in splitReduction()
212 SmallVector<StringRef> reductionIteratorTypes; in splitReduction()
213 SmallVector<AffineExpr> exprs; in splitReduction()
223 SmallVector<AffineMap> reductionMaps = {inputMap, outputMap}; in splitReduction()
284 SmallVector<unsigned> dims; in splitReductionByScaling()
290 SmallVector<int64_t> loopRanges = op.getStaticLoopRanges(); in splitReductionByScaling()
298 SmallVector<Operation *> combinerOps; in splitReductionByScaling()
302 SmallVector<Attribute> neutralElements = llvm::to_vector<4>( in splitReductionByScaling()
329 SmallVector<Value> newOutputs; in splitReductionByScaling()
331 SmallVector<Operation *> initOrAllocTensorOps; in splitReductionByScaling()
332 SmallVector<linalg::FillOp> fillOps; in splitReductionByScaling()
339 SmallVector<Value> dims = in splitReductionByScaling()
358 SmallVector<AffineMap> newMaps; in splitReductionByScaling()
387 SmallVector<StringRef> iteratorTypes = in splitReductionByScaling()
406 SmallVector<LinalgOp> results; in splitReductionByScaling()
415 SmallVector<AffineMap> indexingMaps = { in splitReductionByScaling()
417 SmallVector<StringRef> reductionIteratorTypes( in splitReductionByScaling()