Copy input tensor in a RandomShuffleOp to output directly when its nu…
…mber of elements is less than or equal to 1.

This fixes the crash in TFXLA MLIR bridge when the input tensor has a rank of 0. This also makes its behavoir more consistent with TF core kernel.

PiperOrigin-RevId: 504983211