Add the support of uint8 type for Split op.

Change: 137572042

tensorflow/core/kernels/split_lib_cpu.cc

@@ -40,6 +40,7 @@ void Split<Eigen::ThreadPoolDevice, T>::operator()(
 #define DEFINE_CPU_KERNELS(T) template struct Split<Eigen::ThreadPoolDevice, T>;
 
 TF_CALL_ALL_TYPES(DEFINE_CPU_KERNELS)
+DEFINE_CPU_KERNELS(quint8)
 
 }  // namespace functor
 }  // namespace tensorflow

tensorflow/core/kernels/split_op.cc

@@ -251,6 +251,7 @@ class SplitOpGPU : public SplitOpBase<GPUDevice, T> {
                           SplitOpCPU<type>)
 
 TF_CALL_ALL_TYPES(REGISTER_SPLIT);
+REGISTER_SPLIT(quint8);
 
 #undef REGISTER_SPLIT