Rename variables to simplify refactoring (openxla#2245)

This is part of ongoing efforts to sync reference interpreter closer to
the spec openxla#1049