Searched refs:parallelDim (Results 1 – 1 of 1) sorted by relevance
2091 auto parallelDim = mlir::getAffineDimExpr(nextDim++, ctx); in matchAndRewrite() local2095 srcExprs.insert(srcExprs.begin() + inputFeatureDimension, parallelDim); in matchAndRewrite()2110 parallelDim); in matchAndRewrite()2124 dstExprs.insert(dstExprs.begin() + outputFeatureDimension, parallelDim); in matchAndRewrite()2135 auto parallelDim = mlir::getAffineDimExpr(nextDim++, ctx); in matchAndRewrite() local2138 srcExprs.insert(srcExprs.begin() + inputBatchDimension, parallelDim); in matchAndRewrite()2153 parallelDim); in matchAndRewrite()2168 dstExprs.insert(dstExprs.begin() + outputFeatureDim, parallelDim); in matchAndRewrite()