Searched refs:reshapeOutputShape (Results 1 – 1 of 1) sorted by relevance
48 SmallVectorImpl<int64_t> &reshapeOutputShape) { in computeReshapeOutput() argument53 reshapeOutputShape.assign(higherRank, 1); in computeReshapeOutput()70 reshapeOutputShape[higherRightIndex - 1] = higherRankDim; in computeReshapeOutput()90 reshapeOutputShape[higherLeftIndex] = higherRankDim; in computeReshapeOutput()141 SmallVector<int64_t, 4> reshapeOutputShape; in reshapeLowerToHigher() local143 computeReshapeOutput(outputRankShape, lowerRankShape, reshapeOutputShape); in reshapeLowerToHigher()147 ArrayRef<int64_t>(reshapeOutputShape), reshapeInputType.getElementType()); in reshapeLowerToHigher()151 rewriter.getI64ArrayAttr(reshapeOutputShape)); in reshapeLowerToHigher()