[XLA:GPU] Fix the output shape in the autotune unified redzone buffers.
[XLA:GPU] Fix the output shape in the autotune unified redzone buffers.

After this CL, the output shape will not be a tuple if it contains only a single element. I broke this by accident in cl/626122610.

I also added tests that ensure the correct behavior.
