arm64/mmu: align alloc_init_pte prototype with pmd/pud versions