STT-tensorflow/tensorflow/python/kernel_tests/random
Srinivas Vasudevan 455750f362 Add StatelessParameterizedTruncatedNormal sampler.
This sampler supports broadcasting of its input parameters as well as puts the # samples at the left of the output shape, rather than the right.

PiperOrigin-RevId: 317129622
Change-Id: I4b62ad2e89a9637ae8b30b73af4b662ad0caa943
2020-06-18 10:28:38 -07:00
..
BUILD Add StatelessParameterizedTruncatedNormal sampler. 2020-06-18 10:28:38 -07:00
multinomial_op_big_test.py
multinomial_op_test.py
parameterized_truncated_normal_op_test.py Add StatelessParameterizedTruncatedNormal sampler. 2020-06-18 10:28:38 -07:00
random_binomial_test.py Adds a tf.random.stateless_binomial (stateless analogue to tf.random.Generator.binomial). 2020-02-05 14:07:38 -08:00
random_crop_test.py
random_gamma_test.py Allow np dtypes as args to random_gamma. Previously, failed with AttributeError: type object 'numpy.float32' has no attribute 'as_numpy_dtype' 2019-08-16 09:30:44 -07:00
random_grad_test.py
random_ops_test.py Internal change 2020-05-27 15:22:27 -07:00
random_poisson_test.py
random_shuffle_queue_test.py
stateless_random_ops_test.py Enable more TFRT tests. 2020-05-19 00:24:26 -07:00
util.py Allow RandomBinomial op to broadcast parameters. 2019-11-18 20:18:33 -08:00