Lines Matching defs:resultRank
97 const int64_t resultRank = resultTy.getRank();
112 auto expr = rewriter.getAffineDimExpr(dim + resultRank - sourceRank);
117 return AffineMap::get(/*dimCount=*/resultRank,
127 const int64_t resultRank = resultTy.getRank();
131 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank));
423 int64_t resultRank = resultTy.getRank();
512 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank));
513 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank));
523 createDepthwiseConvCollapseMap(resultRank, reassociationMap, rewriter);
532 getNParallelLoopsAttrs(resultRank),
551 createDepthwiseConvCollapseMap(resultRank, reassociationMap, rewriter);