[xla:gpu] Support dynamic offset size in AddressComputationThunk
[xla:gpu] Support dynamic offset size in AddressComputationThunk

Contrary to the documentation, dynamic-slice and DUS ops may have start indices of type different than S64[]. We need to be more precise when loading the offset from device to host at runtime.
