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

[OpenCL] Removes half concat op registration ()

Browse Source
This commit is contained in:
Luke Iwanski 2017-06-07 01:16:22 +01:00 committed by Jonathan Hseu
parent cfdc22dee6
commit 754e12668c
2 changed files with 4 additions and 2 deletions
tensorflow/core/kernels
concat_lib_cpu.ccconcat_op.cc

2
tensorflow/core/kernels/concat_lib_cpu.cc
View File

@ -95,7 +95,7 @@ void ConcatSYCL(const Eigen::SyclDevice& d,
const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&, \
typename TTypes<T, 2>::Matrix* output);
TF_CALL_GPU_NUMBER_TYPES(REGISTER_SYCL)
TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL)
#undef REGISTER_SYCL
#endif // TENSORFLOW_USE_SYCL

4
tensorflow/core/kernels/concat_op.cc
View File

@ -232,7 +232,8 @@ REGISTER_KERNEL_BUILDER(Name("ConcatV2")
.HostMemory("axis"), \
ConcatV2Op<SYCLDevice, type>)
TF_CALL_GPU_NUMBER_TYPES(REGISTER_SYCL);
TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL);
REGISTER_KERNEL_BUILDER(Name("Concat")
.Device(DEVICE_SYCL)
.TypeConstraint<int32>("T")
@ -248,6 +249,7 @@ REGISTER_KERNEL_BUILDER(Name("ConcatV2")
.HostMemory("axis")
.HostMemory("output"),
ConcatV2Op<CPUDevice, int32>);
#undef REGISTER_SYCL
#endif // TENSORFLOW_USE_SYCL