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

[OpenCL] Extends softmax op to cover double (#10323)

Browse Source
This commit is contained in:
Luke Iwanski 2017-06-01 03:30:28 +01:00 committed by Benoit Steiner
parent a702863e80
commit 9634414007
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -90,6 +90,8 @@ REGISTER_KERNEL_BUILDER(
REGISTER_KERNEL_BUILDER( REGISTER_KERNEL_BUILDER(
Name("Softmax").Device(DEVICE_SYCL).TypeConstraint<float>("T"), Name("Softmax").Device(DEVICE_SYCL).TypeConstraint<float>("T"),
SoftmaxOp<SYCLDevice, float>); SoftmaxOp<SYCLDevice, float>);
REGISTER_KERNEL_BUILDER(
Name("Softmax").Device(DEVICE_SYCL).TypeConstraint<double>("T"),
SoftmaxOp<SYCLDevice, double>);
#endif // TENSORFLOW_USE_SYCL #endif // TENSORFLOW_USE_SYCL
} // namespace tensorflow } // namespace tensorflow