message: Adding ROCm support for the stateless random ops
diff: @@ -200,7 +200,7 @@ TF_CALL_double(REGISTER_CPU);
  TF_CALL_int32(REGISTER_INT_CPU);
  TF_CALL_int64(REGISTER_INT_CPU);
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  TF_CALL_half(REGISTER_GPU);
  TF_CALL_float(REGISTER_GPU);
  @@ -208,7 +208,7 @@ TF_CALL_double(REGISTER_GPU);
  TF_CALL_int32(REGISTER_INT_GPU);
  TF_CALL_int64(REGISTER_INT_GPU);
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  #undef REGISTER
  #undef REGISTER_INT
  