OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[OpenCL] Cleans reshape.cc (#10347)

Browse Source 
* [OpenCL] Cleans reshape.cc

* Removes half and complex numbers.

 Half is extension and complex numbers needs implementation in Eigen first
This commit is contained in:
Luke Iwanski 2017-06-06 22:17:54 +01:00 committed by Jonathan Hseu
parent 8cda8660e1
commit ddd67e3333
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/core/kernels/reshape_op.cc
View File

@ -42,8 +42,12 @@ TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
.TypeConstraint<type>("T") \ .TypeConstraint<type>("T") \
.TypeConstraint<int32>("Tshape"), \ .TypeConstraint<int32>("Tshape"), \
ReshapeOp); ReshapeOp);
TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL); REGISTER_SYCL_KERNEL(float)
#undef REGISTER_SYCL_KERNEL REGISTER_SYCL_KERNEL(double)
REGISTER_SYCL_KERNEL(uint8)
REGISTER_SYCL_KERNEL(int8)
REGISTER_SYCL_KERNEL(int64)
REGISTER_SYCL_KERNEL(uint16)
REGISTER_KERNEL_BUILDER(Name("Reshape") REGISTER_KERNEL_BUILDER(Name("Reshape")
.Device(DEVICE_SYCL) .Device(DEVICE_SYCL)
@ -53,6 +57,7 @@ REGISTER_KERNEL_BUILDER(Name("Reshape")
.TypeConstraint<int32>("T") .TypeConstraint<int32>("T")
.TypeConstraint<int32>("Tshape"), .TypeConstraint<int32>("Tshape"),
ReshapeOp); ReshapeOp);
#undef REGISTER_SYCL_KERNEL
#endif // TENSORFLOW_USE_SYCL #endif // TENSORFLOW_USE_SYCL
#if GOOGLE_CUDA #if GOOGLE_CUDA