mtd: add init_size hook for NAND driver