message: Adding ROCm support for variable ops
diff: @@ -209,7 +209,7 @@ TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL);
  #undef REGISTER_SYCL_KERNEL
  #endif // TENSORFLOW_USE_SYCL
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  // Only register 'Variable' on GPU for the subset of types also supported by
  // 'Assign' (see dense_update_ops.cc.)
  #define REGISTER_GPU_KERNELS(type) \
  @@ -236,6 +236,6 @@ TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL);
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNELS);
  TF_CALL_int64(REGISTER_GPU_KERNELS);
  #undef REGISTER_GPU_KERNELS
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  } // namespace tensorflow
  