Searched refs:stateless_multinomial (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | categorical_op_test.py | 149 x = stateless_random_ops.stateless_multinomial(
166 pure = stateless_random_ops.stateless_multinomial(
185 x = stateless_random_ops.stateless_multinomial(
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | interleave_ops.py | 204 gen_stateless_random_ops.stateless_multinomial(logits, 1, seed=seed),
222 gen_stateless_random_ops.stateless_multinomial(logits, 1, seed=seed),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateless_random_ops.py | 584 def stateless_multinomial(logits, function
668 return gen_stateless_random_ops.stateless_multinomial(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.random.pbtxt | 88 name: "stateless_multinomial"
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 169 functools.partial(wrap, stateless.stateless_multinomial, logits,
|