Commit 7cc2dae9 authored by Bixia Zheng's avatar Bixia Zheng Committed by TensorFlower Gardener
Browse files

[XLA:CPU/GPU] Fix a problem in the random number generator. 

Add primitive type S64 to GetNumberOfElementsPerPhiloxRngSample so that the
random number generator can generate random numbers for signed 64-bit integer.

PiperOrigin-RevId: 207621810
parent 2e7ae340
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