message: Add needed specifications for Split on GPU.
diff: @@ -54,6 +54,7 @@ void SplitCustom<Device, T>::operator()(
  TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);
  TF_CALL_complex64(DEFINE_GPU_KERNELS);
  TF_CALL_complex128(DEFINE_GPU_KERNELS);
  +TF_CALL_int64(DEFINE_GPU_KERNELS);
  TF_CALL_bfloat16(DEFINE_GPU_KERNELS);
  #undef DEFINE_GPU_KERNELS
  @@ -245,6 +246,7 @@ struct SplitVOpGPULaunch {
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
  TF_CALL_complex64(REGISTER_GPU_KERNEL);
  TF_CALL_complex128(REGISTER_GPU_KERNEL);
  +TF_CALL_int64(REGISTER_GPU_KERNEL);
  TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
  #undef REGISTER_GPU_KERNEL
  #define REGISTER_GPU_KERNEL(T) \
  @@ -254,6 +256,7 @@ TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
  TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
  TF_CALL_complex64(REGISTER_GPU_KERNEL);
  TF_CALL_complex128(REGISTER_GPU_KERNEL);
  +TF_CALL_int64(REGISTER_GPU_KERNEL);
  TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
  #undef REGISTER_GPU_KERNEL
  