Lines Matching refs:nloops
133 auto nloops = colTensorShape.size(); in rewriteInIm2Col() local
137 SmallVector<utils::IteratorType> img2colIterators(nloops, parallel); in rewriteInIm2Col()
140 AffineMap::getMultiDimIdentityMap(nloops, context)}; in rewriteInIm2Col()
237 auto nloops = indices.size(); in rewriteInIm2Col() local
252 nloops, utils::IteratorType::parallel); in rewriteInIm2Col()
256 AffineMap::get(nloops, 0, exprs, rewriter.getContext())), in rewriteInIm2Col()
257 AffineMap::getMultiDimIdentityMap(nloops, rewriter.getContext())}; in rewriteInIm2Col()
302 auto nloops = colTensorShape.size(); in rewriteInIm2Col() local
305 nloops, utils::IteratorType::parallel); in rewriteInIm2Col()
308 AffineMap::get(nloops, 0, inputExprs, rewriter.getContext()), in rewriteInIm2Col()
309 AffineMap::getMultiDimIdentityMap(nloops, rewriter.getContext())}; in rewriteInIm2Col()
418 auto nloops = colTensorShape.size(); in rewriteInIm2Col() local
422 SmallVector<utils::IteratorType, 3> img2colIterators(nloops, parallel); in rewriteInIm2Col()
425 AffineMap::getMultiDimIdentityMap(nloops, context)}; in rewriteInIm2Col()
553 auto nloops = colTensorShape.size(); in rewriteInIm2Col() local
557 SmallVector<utils::IteratorType> img2colIterators(nloops, parallel); in rewriteInIm2Col()
560 AffineMap::getMultiDimIdentityMap(nloops, context)}; in rewriteInIm2Col()