Searched refs:higherRankShape (Results 1 – 1 of 1) sorted by relevance
42 computeReshapeOutput(ArrayRef<int64_t> higherRankShape, in computeReshapeOutput() argument46 int64_t higherRank = higherRankShape.size(); in computeReshapeOutput()56 higherRankDim = higherRankShape[i]; in computeReshapeOutput()101 ArrayRef<int64_t> higherRankShape = in reshapeLowerToHigher() local103 (void)higherRankShape; in reshapeLowerToHigher()109 if (computeReshapeOutput(higherRankShape, lowerRankShape, reshapeOutputShape) in reshapeLowerToHigher()120 outputType.getShape().size() != higherRankShape.size()) in reshapeLowerToHigher()