experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add CPU registration for UINT64 Split op.

Browse Source 
PiperOrigin-RevId: 276584569
Change-Id: Ia5200d5e2bfac1f89190dd48c56c980db1a416c2
This commit is contained in:
A. Unique TensorFlower 2019-10-24 15:57:46 -07:00 committed by TensorFlower Gardener
parent 33ee471039
commit 00ea430324
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/split_lib_cpu.cc
View File

@ -43,6 +43,7 @@ void Split<Eigen::ThreadPoolDevice, T, NDims>::operator()(
TF_CALL_ALL_TYPES(DEFINE_CPU_KERNELS)
DEFINE_CPU_KERNELS(quint8)
DEFINE_CPU_KERNELS(uint64)
#ifdef TENSORFLOW_USE_SYCL
template <typename T, int NDims>

1
tensorflow/core/kernels/split_op.cc
View File

@ -404,6 +404,7 @@ class SplitOpSYCL : public SplitOpBase<SYCLDevice, T> {
TF_CALL_ALL_TYPES(REGISTER_SPLIT);
REGISTER_SPLIT(quint8);
REGISTER_SPLIT(uint64);
#undef REGISTER_SPLIT