tree 8fbb1fab4f38051ecf091c840d76ebba6b26eef0
parent 6981355bc1fc9119f22759ec26a4d6b0714e9767
author A. Unique TensorFlower <gardener@tensorflow.org> 1610495968 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1610496312 -0800

Replaces uses of op StatelessRandomGetKeyCounterAlg with StatelessRandomGetKeyCounter and StatelessRandomGetAlg, so that `seed` is no longer required by XLA to be a compile-time constant.

Adds SetIsStateful to StatelessRandomGetKeyCounter and StatelessRandomGetAlg, so that they won't be constant-folded away.

Re-introduces StatelessRandomNormalV2 in tf.random.stateless_normal, since the OOM problem has been fixed.

PiperOrigin-RevId: 351467699
Change-Id: I523482183387ef12e8c3a114570954b1e097a263