[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"), \
|
Name("EluGrad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
|
||||||
EluGradOp<SYCLDevice, type>)
|
EluGradOp<SYCLDevice, type>)
|
||||||
|
|
||||||
REGISTER_SYCL_KERNELS(float);
|
TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNELS);
|
||||||
#undef REGISTER_SYCL_KERNELS
|
#undef REGISTER_SYCL_KERNELS
|
||||||
#endif // TENSORFLOW_USE_SYCL
|
#endif // TENSORFLOW_USE_SYCL
|
||||||
|
|
||||||
|
@ -175,10 +175,6 @@ void EluGradOp<Device, T>::OperateNoTemplate(OpKernelContext* context,
|
|||||||
|
|
||||||
} // namespace tensorflow
|
} // namespace tensorflow
|
||||||
|
|
||||||
#ifdef TENSORFLOW_USE_SYCL
|
|
||||||
#undef EIGEN_USE_SYCL
|
|
||||||
#endif // TENSORFLOW_USE_SYCL
|
|
||||||
|
|
||||||
#undef EIGEN_USE_THREADS
|
#undef EIGEN_USE_THREADS
|
||||||
|
|
||||||
#endif // TENSORFLOW_KERNELS_RELU_OP_H_
|
#endif // TENSORFLOW_KERNELS_RELU_OP_H_
|
||||||
|
Loading…
Reference in New Issue
Block a user