Lines Matching defs:reductionDims

70     ArrayRef<int64_t> reductionDims = multiReductionOp.getReductionDims();
71     llvm::SmallDenseSet<int64_t> reductionDimsSet(reductionDims.begin(),
72                                                   reductionDims.end());
73     int64_t reductionSize = reductionDims.size();
90                              reductionDims.size(),
91                              parallelDims.size() + reductionDims.size()))))
97       indices.append(reductionDims.begin(), reductionDims.end());
99       indices.append(reductionDims.begin(), reductionDims.end());
185     SmallVector<int64_t, 4> reductionDims, reductionShapes;
191         reductionDims.push_back(i);
224     // Check parallelDims are exactly {reductionDims.size()} + [0 .. size).
225     counter = reductionDims.size();