Fix the data_type_transform GPU test #8852
Description
Right now, the data_type_transform test on GPU occasionally fails.
In order to fix this, we need to add additional context.Wait() between tensor copy and transform on GPU.
{{ message }}
Right now, the data_type_transform test on GPU occasionally fails.
In order to fix this, we need to add additional context.Wait() between tensor copy and transform on GPU.