xla

8f121c31 - Patch RNG for better memory utilization in dropout layers (#5710)

Patch RNG for better memory utilization in dropout layers (#5710) * Enable u8 rng-bit-generator with downcast * Use BF16 values if downcast for uniform dist is set.