Searched refs:reducedShape (Results 1 – 2 of 2) sorted by relevance
439 ArrayRef<int64_t> reducedShape) { in computeRankReductionMask() argument440 size_t originalRank = originalShape.size(), reducedRank = reducedShape.size(); in computeRankReductionMask()446 originalShape[originalIdx] == reducedShape[reducedIdx]) { in computeRankReductionMask()
344 ArrayRef<int64_t> reducedShape);