[OpenCL] Extends softmax op to cover double (#10323)
This commit is contained in:
parent
a702863e80
commit
9634414007
@ -90,6 +90,8 @@ REGISTER_KERNEL_BUILDER(
|
||||
REGISTER_KERNEL_BUILDER(
|
||||
Name("Softmax").Device(DEVICE_SYCL).TypeConstraint<float>("T"),
|
||||
SoftmaxOp<SYCLDevice, float>);
|
||||
REGISTER_KERNEL_BUILDER(
|
||||
Name("Softmax").Device(DEVICE_SYCL).TypeConstraint<double>("T"),
|
||||
SoftmaxOp<SYCLDevice, double>);
|
||||
#endif // TENSORFLOW_USE_SYCL
|
||||
|
||||
} // namespace tensorflow
|
||||
|
Loading…
Reference in New Issue
Block a user