Searched refs:multinom (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/polly/lib/External/isl/ |
| D | isl_bernstein.c | 162 isl_vec *multinom = NULL; in extract_coefficients() local
175 multinom = isl_vec_alloc(ctx, n); in extract_coefficients()
176 if (!c || !k || !left || !multinom) in extract_coefficients()
179 isl_int_set_si(multinom->el[0], 1); in extract_coefficients()
186 isl_int_set(multinom->el[1], multinom->el[0]); in extract_coefficients()
194 isl_int_divexact_ui(multinom->el[i], in extract_coefficients()
195 multinom->el[i], j); in extract_coefficients()
201 ctx->one, multinom->el[i]); in extract_coefficients()
215 isl_int_divexact_ui(multinom->el[i], in extract_coefficients()
216 multinom->el[i], k[i]); in extract_coefficients()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | multinomial_test.py | 81 multinom = multinomial.Multinomial(total_count=3., logits=logits)
82 self.assertEqual((1, 3), multinom.probs.get_shape())
83 self.assertEqual((1, 3), multinom.logits.get_shape())
84 self.assertAllClose(p, multinom.probs)
85 self.assertAllClose(logits, multinom.logits)
114 multinom = multinomial.Multinomial(
116 multinom.prob([2., 1, 2]).eval()
117 multinom.prob([3., 0, 2]).eval()
120 multinom.prob([2., 3, 2]).eval()
125 multinom.prob(x).eval(feed_dict={x: [1.0, 2.5, 1.5]})
[all …]
|