Lines Matching refs:outputDim
320 unsigned outputDim = outputExpr.cast<AffineDimExpr>().getPosition(); in isConvolutionInterfaceImpl() local
321 if (inputExprWalker.unConvolvedDims.count(outputDim) && in isConvolutionInterfaceImpl()
322 !filterDims.count(outputDim)) { in isConvolutionInterfaceImpl()
324 if (*std::next(iteratorTypesRange.begin(), outputDim) != in isConvolutionInterfaceImpl()
327 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()
330 if (inputExprWalker.convolvedDims.count(outputDim) && in isConvolutionInterfaceImpl()
331 !filterDims.count(outputDim)) { in isConvolutionInterfaceImpl()
333 if (*std::next(iteratorTypesRange.begin(), outputDim) != in isConvolutionInterfaceImpl()
336 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()
339 if (!inputExprWalker.convolvedDims.count(outputDim) && in isConvolutionInterfaceImpl()
340 !inputExprWalker.unConvolvedDims.count(outputDim) && in isConvolutionInterfaceImpl()
341 filterDims.count(outputDim)) { in isConvolutionInterfaceImpl()
343 if (*std::next(iteratorTypesRange.begin(), outputDim) != in isConvolutionInterfaceImpl()
346 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()
349 if (inputExprWalker.unConvolvedDims.count(outputDim) && in isConvolutionInterfaceImpl()
350 filterDims.count(outputDim)) { in isConvolutionInterfaceImpl()
352 if (*std::next(iteratorTypesRange.begin(), outputDim) != in isConvolutionInterfaceImpl()
355 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()