tf.random.stateless_uniform