Lines Matching refs:SmallVector
38 static SmallVector<int64_t> getTiledSliceDims(OpOperand *consumerOperand, in getTiledSliceDims()
65 static SmallVector<int64_t>
77 producerIndexingMap.getSubMap(SmallVector<unsigned>( in getTiledProducerLoops()
89 SmallVector<int64_t> tiledProducerLoopIndices; in getTiledProducerLoops()
144 SmallVector<Value> producerLoopBounds; in getTiledProducer()
148 SmallVector<Range> sliceOpRanges = sliceOp.getOrCreateRanges(b, loc); in getTiledProducer()
154 SmallVector<Value> tileIvs(producerOp.getNumLoops(), nullptr); in getTiledProducer()
155 SmallVector<Value> tileSizes(producerOp.getNumLoops(), zero); in getTiledProducer()
156 SmallVector<Value> allIvs(producerOp.getNumLoops(), nullptr); in getTiledProducer()
165 SmallVector<Value> tiledOperands = producerOp.getInputAndOutputOperands(); in getTiledProducer()
220 SmallVector<BlockArgument> TileLoopNest::getTiedBBArgs(BlockArgument bbArg) { in getTiedBBArgs()
222 SmallVector<BlockArgument> bbArgs; in getTiedBBArgs()
239 SmallVector<BlockArgument> bbArgs = getTiedBBArgs(bbArg); in getTiedIterArg()
269 SmallVector<BlockArgument> bbArgs = getTiedBBArgs(bbArg); in hasOtherUses()
286 .setInterchange(SmallVector<unsigned>( in tileRootOp()
373 SmallVector<int64_t> tiledSliceDimIndices = getTiledSliceDims( in fuseProducer()
379 SmallVector<int64_t> tiledProducerLoopIndices = in fuseProducer()
408 SmallVector<LinalgOp> TileLoopNest::getAllTiledAndFusedOps() { in getAllTiledAndFusedOps()
409 SmallVector<LinalgOp> result; in getAllTiledAndFusedOps()
437 SmallVector<StringAttr> iterTypes = in tileConsumerAndFuseProducers()
447 SmallVector<OpOperand *> candidates(operands.begin(), operands.end()); in tileConsumerAndFuseProducers()
461 SmallVector<int64_t> outerTileSizes(tileSizes.size(), 0); in tileConsumerAndFuseProducers()
462 SmallVector<int64_t> innerTileSizes(tileSizes.size(), 0); in tileConsumerAndFuseProducers()