Home
last modified time | relevance | path

Searched refs:Multinomial (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dmultinomial_test.py41 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(
138 pmf = multinomial.Multinomial(total_count=1., probs=p).prob(counts)
[all …]
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_Multinomial.pbtxt2 graph_op_name: "Multinomial"
4 name: "random.Multinomial"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DMultinomial.pbtxt2 name: "Multinomial"
49 name: "Multinomial"
98 name: "Multinomial"
161 name: "Multinomial"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DMultinomial.pbtxt2 name: "Multinomial"
49 name: "Multinomial"
98 name: "Multinomial"
161 name: "Multinomial"
/external/tensorflow/tensorflow/core/kernels/
Dmultinomial_op_test.cc27 static Graph* Multinomial(int batch_size, int num_classes, int num_samples) { in Multinomial() function
46 test::Benchmark(#DEVICE, Multinomial(B, C, S), \
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distributions.-multinomial.pbtxt1 path: "tensorflow.distributions.Multinomial"
3 is_instance: "<class \'tensorflow.python.ops.distributions.multinomial.Multinomial\'>"
53 …varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\', \'True\', \'Multinomial\'], "
Dtensorflow.distributions.pbtxt44 name: "Multinomial"
Dtensorflow.raw_ops.pbtxt2576 name: "Multinomial"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_Multinomial.pbtxt2 graph_op_name: "Multinomial"
/external/tensorflow/tensorflow/python/ops/distributions/
Dmultinomial.py56 class Multinomial(distribution.Distribution): class
208 super(Multinomial, self).__init__(
Ddistributions.py35 from tensorflow.python.ops.distributions.multinomial import Multinomial
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Multinomial.pbtxt2 graph_op_name: "Multinomial"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir292 // CHECK-NOT: tf.Multinomial
294 …%1 = "tf.Multinomial"(%arg0, %samples) {seed = 0, seed2 = 0}: (tensor<2x4xf32>, tensor<i32>) -> te…
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md283 * `raw_ops.Multinomial`
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md149 `Multinomial` | `output_dtype={int32,int64}`<br>`T={double,float,int32,int6…
Dcpu_supported_ops.md149 `Multinomial` | `output_dtype={int32,int64}`<br>`T={double,float,int32,int6…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt2576 name: "Multinomial"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt18347 name: "Multinomial"
18393 name: "Multinomial"
Dops_history.v2.pbtxt39880 name: "Multinomial"
39927 name: "Multinomial"
39976 name: "Multinomial"
40039 name: "Multinomial"
/external/tensorflow/
DRELEASE.md3453 * Ensure `tf.distributions.Multinomial` doesn't underflow in `log_prob`.
/external/tensorflow/tensorflow/go/op/
Dwrappers.go29668 func Multinomial(scope *Scope, logits tf.Output, num_samples tf.Output, optional ...MultinomialAttr… func
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td8696 def TF_MultinomialOp : TF_Op<"Multinomial", [TF_CannotDuplicate]> {
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt25686 name: "Multinomial"