Searched refs:nloops (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConvertConv2DToImg2Col.cpp | 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()
[all …]
|
| H A D | ElementwiseOpFusion.cpp | 219 unsigned nloops, llvm::SmallDenseSet<int> &preservedProducerResults) { in generateFusedElementwiseOpRegion()
174 generateFusedElementwiseOpRegion(RewriterBase & rewriter,GenericOp fusedOp,AffineMap consumerToProducerLoopsMap,OpOperand * fusedOperand,unsigned nloops,llvm::SmallDenseSet<int> & preservedProducerResults) generateFusedElementwiseOpRegion() argument
|
| /llvm-project/llvm/test/CodeGen/Hexagon/ |
| H A D | vgather-opt-addr.ll | 36 define dso_local void @contiguos_vgather_test(i32 %Rb, i32 %mu, i32 %nloops, <32 x i32> %Vv, <64 x i32> %Vvv, <32 x i32> %Qs) local_unnamed_addr #0 {
40 %cmp23 = icmp sgt i32 %nloops, 0
65 %exitcond = icmp eq i32 %inc, %nloops
|