Searched refs:oldShape (Results 1 – 4 of 4) sorted by relevance
113 ArrayRef<int64_t> oldShape = allocOp.getType().getShape(); in multiBuffer() local114 newShape.append(oldShape.begin(), oldShape.end()); in multiBuffer()132 offsets.append(oldShape.size(), builder.getIndexAttr(0)); in multiBuffer()133 strides.assign(oldShape.size() + 1, builder.getIndexAttr(1)); in multiBuffer()135 for (int64_t size : oldShape) in multiBuffer()
73 ArrayRef<int64_t> oldShape = oldMemRefType.getShape(); in doubleBuffer() local76 std::copy(oldShape.begin(), oldShape.end(), newShape.begin() + 1); in doubleBuffer()
25 ArrayRef<int64_t> oldShape = oldType.getShape(); in trimLeadingOneDims() local27 oldShape.drop_while([](int64_t dim) { return dim == 1; }); in trimLeadingOneDims()30 newShape = oldShape.take_back(); in trimLeadingOneDims()
160 ArrayRef<int64_t> oldShape = op.getShape(op.getOutputOperand(0)); in splitReduction() local170 newOutputShape.push_back(oldShape[oldDim]); in splitReduction()