It's clearer to have one root partition size variable that is used in
the setup of both A and B partitions.

Signed-off-by: Andrei Gherzan <andrei.gherzan@huawei.com>