Home
last modified time | relevance | path

Searched refs:multinomial (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dmultinomial_test.py28 from tensorflow.python.ops.distributions import multinomial
41 dist = multinomial.Multinomial(total_count=1., probs=p)
52 dist = multinomial.Multinomial(total_count=n, probs=p)
63 dist = multinomial.Multinomial(total_count=n, probs=p)
71 dist = multinomial.Multinomial(total_count=3., probs=p)
81 multinom = multinomial.Multinomial(total_count=3., logits=logits)
91 dist = multinomial.Multinomial(total_count=1., logits=logits)
100 dist = multinomial.Multinomial(total_count=n, probs=p, validate_args=True)
114 multinom = multinomial.Multinomial(
127 multinom = multinomial.Multinomial(
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/random/
Dmultinomial_op_test.py52 native_sampler = random_ops.multinomial
65 samples = self.evaluate(random_ops.multinomial(
105 samples = self.evaluate(random_ops.multinomial(logits, 10))
142 sample_op1 = random_ops.multinomial(logits, num_samples, seed)
143 sample_op2 = random_ops.multinomial(logits, num_samples, seed)
195 random_ops.multinomial(
202 x = random_ops.multinomial(array_ops.zeros([5, 0]), 7)
211 samples = self.evaluate(random_ops.multinomial(logits, num_samples))
Dmultinomial_op_big_test.py38 samples = random_ops.multinomial(
56 samples = random_ops.multinomial(
76 samples = random_ops.multinomial(
Dstateless_random_ops_test.py171 functools.partial(wrap, random_ops.multinomial, logits,
/external/tensorflow/tensorflow/lite/kernels/
Dmultinomial.cc28 namespace multinomial { namespace
199 static TfLiteRegistration r = {multinomial::Init, multinomial::Free, in Register_MULTINOMIAL()
200 multinomial::Prepare, multinomial::Eval}; in Register_MULTINOMIAL()
DBUILD735 "multinomial.cc",
/external/tensorflow/tensorflow/compiler/tests/
Dcategorical_op_test.py62 op = random_ops.multinomial(logits, num_samples,
101 return random_ops.multinomial(np.array([[1., 1., 1.]], dtype=dtype), 10,
113 x = random_ops.multinomial(
176 x = random_ops.multinomial(
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StatelessMultinomial.pbtxt29 summary: "Draws samples from a multinomial distribution."
Dapi_def_Multinomial.pbtxt36 summary: "Draws samples from a multinomial distribution."
/external/tensorflow/tensorflow/python/ops/distributions/
Ddistributions.py35 from tensorflow.python.ops.distributions.multinomial import Multinomial
Ddirichlet_multinomial.py271 draws = random_ops.multinomial(
Dmultinomial.py263 x = random_ops.multinomial(logits[array_ops.newaxis, ...], n_draw,
Dcategorical.py277 draws = random_ops.multinomial(
/external/tensorflow/tensorflow/python/ops/
Drandom_ops.py472 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): function
532 return gen_random_ops.multinomial(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distributions.-multinomial.pbtxt3 is_instance: "<class \'tensorflow.python.ops.distributions.multinomial.Multinomial\'>"
Dtensorflow.random.pbtxt52 name: "multinomial"
Dtensorflow.pbtxt1712 name: "multinomial"
/external/rust/crates/num-integer/src/
Dlib.rs1127 pub fn multinomial<T: Integer + Clone>(k: &[T]) -> T in multinomial() function
1278 assert_eq!(multinomial(k), binomial(n, k[0])); in test_multinomial()
1312 assert_eq!(multinomial(k), expected); in test_multinomial()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir290 // CHECK-LABEL: multinomial
291 func @multinomial(%arg0: tensor<2x4xf32>, %seed: tensor<i32>, %seed2: tensor<i32>) -> tensor<2x10xi…
/external/tensorflow/tensorflow/tools/ci_build/
Dpylint_allowlist96 ^tensorflow/python/ops/distributions/multinomial.py.*\[E1130.*invalid-unary-operand-type
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py3138 samples = gen_random_ops.multinomial(
3146 samples = gen_random_ops.multinomial(
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td8697 let summary = "Draws samples from a multinomial distribution.";
14749 let summary = "Draws samples from a multinomial distribution.";
/external/tensorflow/
DRELEASE.md3759 * Fix incorrect sampling of small probabilities in CPU/GPU multinomial.
/external/one-true-awk/testdir/
Dfunstack.in17228 …arable closed queueing networks requires the evaluation of homogeneous multinomial expressions. Th…
17238 …operation count; expressions; homogeneous multinomial; Horner's rule; load dependent service rate;…