Searched refs:reshapeOutputShape (Results 1 – 1 of 1) sorted by relevance
44 SmallVectorImpl<int64_t> &reshapeOutputShape) { in computeReshapeOutput() argument49 reshapeOutputShape.assign(higherRank, 1); in computeReshapeOutput()60 reshapeOutputShape[i] = 1; in computeReshapeOutput()63 reshapeOutputShape[i] = lowerRankDim; in computeReshapeOutput()107 SmallVector<int64_t, 4> reshapeOutputShape; in reshapeLowerToHigher() local109 if (computeReshapeOutput(higherRankShape, lowerRankShape, reshapeOutputShape) in reshapeLowerToHigher()115 ArrayRef<int64_t>(reshapeOutputShape), reshapeInputType.getElementType()); in reshapeLowerToHigher()119 if (outputType.getShape().size() != reshapeOutputShape.size() || in reshapeLowerToHigher()126 rewriter.getI64ArrayAttr(reshapeOutputShape)); in reshapeLowerToHigher()