tf.random.uniform_candidate_sampler