message: Adding uint32 support for more variable related operations.
diff: @@ -282,6 +282,7 @@ REGISTER_KERNEL_BUILDER(
  TF_CALL_GPU_ALL_TYPES(REGISTER_GPU_KERNELS);
  TF_CALL_int64(REGISTER_GPU_KERNELS);
  TF_CALL_variant(REGISTER_GPU_KERNELS);
  +TF_CALL_uint32(REGISTER_GPU_KERNELS);
  #undef REGISTER_GPU_KERNELS
  REGISTER_KERNEL_BUILDER(Name("_VarHandlesOp")
  @@ -511,6 +512,7 @@ class AssignVariableOp<Device, Variant> : public OpKernel {
  TF_CALL_ALL_TYPES(REGISTER_KERNELS);
  TF_CALL_QUANTIZED_TYPES(REGISTER_KERNELS);
  +TF_CALL_uint32(REGISTER_KERNELS);
  #undef REGISTER_KERNELS
  #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  @@ -524,6 +526,7 @@ TF_CALL_QUANTIZED_TYPES(REGISTER_KERNELS);
  TF_CALL_GPU_ALL_TYPES(REGISTER_GPU_KERNELS);
  TF_CALL_int64(REGISTER_GPU_KERNELS);
  TF_CALL_variant(REGISTER_GPU_KERNELS);
  +TF_CALL_uint32(REGISTER_GPU_KERNELS);
  #undef REGISTER_GPU_KERNELS
  #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  