message: Fix int64 failure on GPU for TensorArray
  This fix tries to address the issue raised in 22054 where
  int64 on GPU results in colocation errors. This fix enables
  int64 on GPU with TensorArray.
  This fix fixes 22054.
diff: @@ -259,6 +259,7 @@ REGISTER_KERNEL_BUILDER(Name("TensorArrayV3").Device(DEVICE_CPU),
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU);
  TF_CALL_complex64(REGISTER_GPU);
  TF_CALL_complex128(REGISTER_GPU);
  +TF_CALL_int64(REGISTER_GPU);
  REGISTER_GPU(bfloat16);
  #undef REGISTER_GPU
  @@ -487,6 +488,7 @@ TF_CALL_ALL_TYPES(REGISTER_WRITE);
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU);
  TF_CALL_complex64(REGISTER_GPU);
  TF_CALL_complex128(REGISTER_GPU);
  +TF_CALL_int64(REGISTER_GPU);
  REGISTER_GPU(bfloat16);
  #undef REGISTER_GPU
  @@ -576,6 +578,7 @@ TF_CALL_ALL_TYPES(REGISTER_READ)
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU);
  TF_CALL_complex64(REGISTER_GPU);
  TF_CALL_complex128(REGISTER_GPU);
  +TF_CALL_int64(REGISTER_GPU);
  REGISTER_GPU(bfloat16);
  #undef REGISTER_GPU
  @@ -777,6 +780,7 @@ REGISTER_GATHER_AND_PACK(qint32);
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU);
  TF_CALL_complex64(REGISTER_GPU);
  TF_CALL_complex128(REGISTER_GPU);
  +TF_CALL_int64(REGISTER_GPU);
  REGISTER_GPU(bfloat16);
  #undef REGISTER_GPU
  @@ -998,6 +1002,7 @@ REGISTER_CONCAT(qint32);
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU);
  TF_CALL_complex64(REGISTER_GPU);
  TF_CALL_complex128(REGISTER_GPU);
  +TF_CALL_int64(REGISTER_GPU);
  REGISTER_GPU(bfloat16);
  #undef REGISTER_GPU
  @@ -1218,6 +1223,7 @@ TF_CALL_ALL_TYPES(REGISTER_SCATTER_AND_UNPACK);
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU);
  TF_CALL_complex64(REGISTER_GPU);
  TF_CALL_complex128(REGISTER_GPU);
  +TF_CALL_int64(REGISTER_GPU);
  #undef REGISTER_GPU
  #endif // GOOGLE_CUDA
  @@ -1388,6 +1394,7 @@ TF_CALL_ALL_TYPES(REGISTER_SPLIT);
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU);
  TF_CALL_complex64(REGISTER_GPU);
  TF_CALL_complex128(REGISTER_GPU);
  +TF_CALL_int64(REGISTER_GPU);
  #undef REGISTER_GPU
  #endif // GOOGLE_CUDA
  