Searched refs:outputRank (Results 1 – 1 of 1) sorted by relevance
150 int64_t outputRank, SmallVector<ReassociationExprs, 4> &reassociationMap, in createDepthwiseConvCollapseMap() argument152 reassociationMap.resize(outputRank); in createDepthwiseConvCollapseMap()153 for (int i = 0; i < outputRank; i++) { in createDepthwiseConvCollapseMap()156 reassociationMap[outputRank - 1].push_back( in createDepthwiseConvCollapseMap()157 rewriter.getAffineDimExpr(outputRank)); in createDepthwiseConvCollapseMap()