Small polish of GPU stream/event synchronize kernel implementations.
Small polish of GPU stream/event synchronize kernel implementations.

PiperOrigin-RevId: 372966914

tensorflow/runtime,tensorflow/runtime,tensorflow/runtime,tensorflow/runtime