Searched refs:higherRankShape (Results 1 – 1 of 1) sorted by relevance
46 static void computeReshapeOutput(ArrayRef<int64_t> higherRankShape, in computeReshapeOutput() argument50 int64_t higherRank = higherRankShape.size(); in computeReshapeOutput()65 higherRankDim = higherRankShape[higherRightIndex - 1]; in computeReshapeOutput()85 higherRankDim = higherRankShape[higherLeftIndex]; in computeReshapeOutput()132 ArrayRef<int64_t> higherRankShape = in reshapeLowerToHigher() local134 (void)higherRankShape; in reshapeLowerToHigher()139 assert(higherRankShape.size() == outputRankShape.size()); in reshapeLowerToHigher()