Commit e0f5241e authored by David Majnemer's avatar David Majnemer Committed by TensorFlower Gardener
Browse files

[TF:XLA] Don't let RNG yield 0 for the gumbel softmax trick 

We take the log(-log(x)), as such we would not want to let x be 0 or 1 as we
would get a non-finite number.

This was found by inspection, I have no evidence that this happens in practice.

PiperOrigin-RevId: 234813142
parent 134915e5
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment