xla

fa5d132c - Improve bernoulli rng-bit-generation memory footprint (#5581) (#5589)

Improve bernoulli rng-bit-generation memory footprint (#5581) (#5589) * Allow downcasting RngUniform genenration for Bernoulli Co-authored-by: Yeounoh Chung <yeounoh@google.com>