Move `diff_dict` to compute API

`diff_dict` is only once, in the compute API, so just move it there.

Change-Id: I3a5a56a969aa661b096587b82fd4dc3fc430995f
Merge "Move `diff_dict` to compute API"