HomeSort by relevance Sort by last modified time
    Searched refs:stateless_multinomial (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/stateless/python/kernel_tests/
stateless_random_ops_test.py 38 self.assertIs(stateless.stateless_multinomial,
39 stateless_random_ops.stateless_multinomial)
  /external/tensorflow/tensorflow/contrib/stateless/
__init__.py 29 @@stateless_multinomial
42 from tensorflow.python.ops.stateless_random_ops import stateless_multinomial
  /external/tensorflow/tensorflow/compiler/tests/
categorical_op_test.py 148 x = stateless_random_ops.stateless_multinomial(
165 pure = stateless_random_ops.stateless_multinomial(
184 x = stateless_random_ops.stateless_multinomial(
  /external/tensorflow/tensorflow/python/data/experimental/ops/
interleave_ops.py 184 # The `stateless_multinomial()` op expects log-probabilities, as opposed
196 gen_stateless_random_ops.stateless_multinomial(logits, 1, seed=seed),
208 # The `stateless_multinomial()` op expects log-probabilities, as opposed to
214 gen_stateless_random_ops.stateless_multinomial(logits, 1, seed=seed),
  /external/tensorflow/tensorflow/python/ops/
stateless_random_ops.py 183 @tf_export(v1=["random.stateless_multinomial"])
186 def stateless_multinomial(logits, function
204 samples = tf.random.stateless_multinomial(
219 with ops.name_scope(name, "stateless_multinomial", [logits, seed]):
266 return gen_stateless_random_ops.stateless_multinomial(
  /external/tensorflow/tensorflow/python/kernel_tests/random/
stateless_random_ops_test.py 129 yield (functools.partial(stateless.stateless_multinomial, **kwds),

Completed in 167 milliseconds