[OpenCL] Cleans relu ops (#10343)
* [OpenCL] register relu ops to gpu types (no half) * [OpenCL] Removes #undef EIGEN_USE_SYCL
This commit is contained in:
parent
09c1455e3a
commit
9fd7cf0547
@ -156,7 +156,7 @@ TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNELS);
|
||||
Name("EluGrad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
|
||||
EluGradOp<SYCLDevice, type>)
|
||||
|
||||
REGISTER_SYCL_KERNELS(float);
|
||||
TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNELS);
|
||||
#undef REGISTER_SYCL_KERNELS
|
||||
#endif // TENSORFLOW_USE_SYCL
|
||||
|
||||
|
@ -175,10 +175,6 @@ void EluGradOp<Device, T>::OperateNoTemplate(OpKernelContext* context,
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#ifdef TENSORFLOW_USE_SYCL
|
||||
#undef EIGEN_USE_SYCL
|
||||
#endif // TENSORFLOW_USE_SYCL
|
||||
|
||||
#undef EIGEN_USE_THREADS
|
||||
|
||||
#endif // TENSORFLOW_KERNELS_RELU_OP_H_
|
||||
|
Loading…
Reference in New Issue
Block a user