search

lukeiwanski / tensorflow

OpenCL support for TensorFlow via SYCL
Apache License 2.0
65 stars 14 forks source link

[SYCL] MultinomialOp uses double internally #223

Closed Rbiessy closed 6 years ago

Rbiessy commented 6 years ago

Cast to double if possible in case of low probabilities