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

[OpenCL] Registers RGBToHSV and HSVToRGB () ()

Browse Source 
* [OpenCL] Added RGBToHSV and HSVToRGB

* Aligning '\'
This commit is contained in:
Luke Iwanski 2017-06-22 01:00:43 +01:00 committed by Benoit Steiner
parent 832894ef89
commit fc7361081f
1 changed files with 15 additions and 0 deletions

15
tensorflow/core/kernels/colorspace_op.cc
View File

@ -35,6 +35,9 @@ namespace tensorflow {
typedef Eigen::ThreadPoolDevice CPUDevice; typedef Eigen::ThreadPoolDevice CPUDevice;
typedef Eigen::GpuDevice GPUDevice; typedef Eigen::GpuDevice GPUDevice;
#ifdef TENSORFLOW_USE_SYCL
typedef Eigen::SyclDevice SYCLDevice;
#endif
template <typename Device, typename T> template <typename Device, typename T>
class RGBToHSVOp : public OpKernel { class RGBToHSVOp : public OpKernel {
@ -146,4 +149,16 @@ TF_CALL_float(REGISTER_GPU);
TF_CALL_double(REGISTER_GPU); TF_CALL_double(REGISTER_GPU);
#endif #endif
#ifdef TENSORFLOW_USE_SYCL
#define REGISTER_SYCL(T) \
REGISTER_KERNEL_BUILDER(Name("RGBToHSV").Device(DEVICE_SYCL) \
.TypeConstraint<T>("T"), \
RGBToHSVOp<SYCLDevice, T>); \
REGISTER_KERNEL_BUILDER(Name("HSVToRGB").Device(DEVICE_SYCL) \
.TypeConstraint<T>("T"), \
HSVToRGBOp<SYCLDevice, T>);
TF_CALL_float(REGISTER_SYCL);
TF_CALL_double(REGISTER_SYCL);
#endif
} // namespace tensorflow } // namespace tensorflow