Lines Matching defs:reductionMask

112     SmallVector<bool> reductionMask(srcRank, false);
115         reductionMask[srcRank - i - 1] = true;
117         reductionMask[i] = true;
122         multiReductionOp.getAcc(), reductionMask, multiReductionOp.getKind());
178     SmallVector<bool> reductionMask = multiReductionOp.getReductionMask();
179     if (srcRank == 2 && reductionMask.front() != reductionMask.back())
187     for (const auto &it : llvm::enumerate(reductionMask)) {
447     SmallVector<bool, 2> reductionMask{false, true};
465         loc, cast, castAcc, reductionMask, multiReductionOp.getKind());