Many of the mm functions need minor tweaks to work with arm64:
- reading 32-bit ints instead of 64-bit words
- using proper page offsets
- vmemmap instead of sparsemem

Update these functions.

Change-Id: I407fdf84c64dbfed870a95fd439bef28598714cb