Searched refs:nLoops (Results 1 – 6 of 6) sorted by relevance
320 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl() local322 tileSizes = tileSizes.take_front(nLoops); in tileLinalgOpImpl()487 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl() local490 if (tileSizeVector.size() < nLoops) { in tileLinalgOpImpl()492 tileSizeVector.append(nLoops - tileSizeVector.size(), zero); in tileLinalgOpImpl()
469 unsigned nLoops = allIvs.size(); in emitScalarImplementation() local471 indexedValues.reserve(nLoops + nInputs + nOutputs); in emitScalarImplementation()472 for (unsigned i = 0; i < nLoops; ++i) in emitScalarImplementation()
223 for (unsigned i = 0, nLoops = loopRanges.size(); i < nLoops; ++i) { in fuse() local
224 unsigned nLoops = iteratorTypes.size(); in build() local225 SmallVector<Type, 4> blockArgTypes(nLoops, builder.getIndexType()); in build()234 bodyBlock->getArguments().take_front(nLoops), in build()235 bodyBlock->getArguments().drop_front(nLoops)); in build()426 auto nLoops = op.getNumLoops(); in verify() local428 if (block.getNumArguments() != nOperands + nLoops) in verify()434 for (unsigned i = 0; i < nLoops; ++i) in verify()440 unsigned memrefArgIndex = i + nLoops; in verify()500 auto nLoops = op.getNumLoops(); in verifyGenericOp() local530 if (m.getNumDims() != nLoops) in verifyGenericOp()[all …]
218 unsigned nLoops = iteratorTypes.size(); local220 nLoops - iteratorTypes.drop_while(isParallelIteratorType).size();
56 unsigned nLoops, unsigned nReduction) { in GetParallelAndReductionIterators() argument57 SmallVector<StringRef, 3> res(nLoops - nReduction, in GetParallelAndReductionIterators()