Searched defs:reductionMask (Results 1 – 3 of 3) sorted by relevance
115 reductionMask[srcRank - i - 1] = true; in matchAndRewrite() local 181 SmallVector<bool> reductionMask = multiReductionOp.getReductionMask(); matchAndRewrite() local 451 SmallVector<bool, 2> reductionMask{false, true}; matchAndRewrite() local [all...]
150 SmallVector<bool> reductionMask = reduceOp.getReductionMask(); in matchAndRewrite() local
443 build(OpBuilder & builder,OperationState & result,Value source,Value acc,ArrayRef<bool> reductionMask,CombiningKind kind) build() argument