Lines Matching defs:inputExprWalker

699 /// subcomputation, as captured by `inputExprWalker`. If
707                          ConvAccessExprWalker &inputExprWalker,
719   llvm::SmallDenseSet<int64_t> batch = inputExprWalker.unConvolvedDims;
724   llvm::SmallDenseSet<int64_t> oi = inputExprWalker.convolvedDims;
730   llvm::set_subtract(oc, inputExprWalker.unConvolvedDims);
735   llvm::set_intersect(depth, inputExprWalker.unConvolvedDims);
742   llvm::SmallDenseSet<int64_t> fl = inputExprWalker.convolvedDims;
746   llvm::SmallDenseSet<int64_t> ic = inputExprWalker.unConvolvedDims;
774       strideExprs.push_back(inputExprWalker.strideAndDilationMapping[oiDim]);
784       dilationExprs.push_back(inputExprWalker.strideAndDilationMapping[flDim]);
825   ConvAccessExprWalker inputExprWalker;
827     (void)inputExprWalker.visit(expr);
828   inputExprWalker.clearMultiUseDims(indexingMaps[0]);
830   return inferConvolutionDimsImpl(linalgOp, inputExprWalker,
861   ConvAccessExprWalker inputExprWalker;
863                    [&inputExprWalker](AffineExpr expr) {
864                      return failed(inputExprWalker.visit(expr));
895     if (inputExprWalker.unConvolvedDims.count(outputDim) &&
903     if (inputExprWalker.convolvedDims.count(outputDim) &&
911     if (!inputExprWalker.convolvedDims.count(outputDim) &&
912         !inputExprWalker.unConvolvedDims.count(outputDim) &&
920     if (inputExprWalker.unConvolvedDims.count(outputDim) &&
933         !inputExprWalker.unConvolvedDims.count(filterDim) &&
934         !inputExprWalker.convolvedDims.count(filterDim)) {
938     if (inputExprWalker.convolvedDims.count(filterDim) &&
948     if (inputExprWalker.unConvolvedDims.count(filterDim) &&
958     if (inputExprWalker.unConvolvedDims.count(filterDim) &&
969   if (!allowEmptyConvolvedDims && inputExprWalker.convolvedDims.empty())
974         linalgOp, inputExprWalker, allowEmptyConvolvedDims);