xla

0daf6fe9 - Add a api to reset and extract RNG Seed (#4196)

Add a api to reset and extract RNG Seed (#4196) * Add a api to reset and extract RNG Seed * rename to _get_rng_seed_as_tensor and make reset optional * minor logic fix * add a TODO