Searched refs:reducedShape (Results 1 – 2 of 2) sorted by relevance
316 ArrayRef<int64_t> reducedShape);
3182 ArrayRef<int64_t> reducedShape) { in computeRankReductionMask() argument3183 size_t originalRank = originalShape.size(), reducedRank = reducedShape.size(); in computeRankReductionMask()3190 (originalShape[originalIdx] == reducedShape[reducedIdx]); in computeRankReductionMask()3234 ArrayRef<int64_t> reducedShape = reducedShapedType.getShape(); in isRankReducedType() local3236 reducedRank = reducedShape.size(); in isRankReducedType()3240 auto optionalMask = computeRankReductionMask(originalShape, reducedShape); in isRankReducedType()