Lines Matching refs:numLoops
57 CodeGen(SparsificationOptions o, unsigned numTensors, unsigned numLoops, in CodeGen()
59 : options(o), loops(numLoops), sizes(numLoops), buffers(numTensors), in CodeGen()
60 pointers(numTensors, std::vector<Value>(numLoops)), in CodeGen()
61 indices(numTensors, std::vector<Value>(numLoops)), in CodeGen()
62 highs(numTensors, std::vector<Value>(numLoops)), in CodeGen()
63 pidxs(numTensors, std::vector<Value>(numLoops)), in CodeGen()
64 idxs(numTensors, std::vector<Value>(numLoops)), redVal(), sparseOut(op), in CodeGen()
337 unsigned numLoops = iteratorTypes.size(); in isAdmissableTensorExp() local
338 for (unsigned i = 0; i < numLoops; i++) in isAdmissableTensorExp()
354 for (unsigned i = 0; i < numLoops; i++) { in isAdmissableTensorExp()
1721 unsigned numLoops = op.iterator_types().getValue().size(); in matchAndRewrite() local
1722 Merger merger(numTensors, numLoops); in matchAndRewrite()
1754 CodeGen codegen(options, numTensors, numLoops, sparseOut, outerParNest); in matchAndRewrite()