Searched refs:outputDims (Results 1 – 2 of 2) sorted by relevance
302 llvm::SmallDenseSet<unsigned> outputDims = in isConvolutionInterfaceImpl() local362 if (outputDims.count(filterDim) && in isConvolutionInterfaceImpl()369 !outputDims.count(filterDim)) { in isConvolutionInterfaceImpl()380 !outputDims.count(filterDim)) { in isConvolutionInterfaceImpl()391 outputDims.count(filterDim)) { in isConvolutionInterfaceImpl()590 llvm::SmallBitVector outputDims(resultShapesFromInputShapesMap.getNumDims()); in reifyResultShapes() local591 outputDims.set(resultShapesSubMapPos.first, resultShapesSubMapPos.second); in reifyResultShapes()592 HasAffineDimExprVisitor checkDimExpr(std::move(outputDims)); in reifyResultShapes()
159 auto outputDims = getDimPositions(genericOp.getIndexingMapsArray().back()); in matchAndRewrite() local193 dstIndices[dim] = indices[outputDims[dim]]; in matchAndRewrite()