Lines Matching refs:du
39 du = distribution_util
68 with ops.control_dependencies([du.assert_close(x, y)]):
71 with ops.control_dependencies([du.assert_close(y, x)]):
75 with ops.control_dependencies([du.assert_close(x, z)]):
79 with ops.control_dependencies([du.assert_close(y, z)]):
88 with ops.control_dependencies([du.assert_close(x, y)]):
91 with ops.control_dependencies([du.assert_close(y, x)]):
95 with ops.control_dependencies([du.assert_close(x, z)]):
99 with ops.control_dependencies([du.assert_close(y, z)]):
109 with ops.control_dependencies([du.assert_close(x, z)]):
113 with ops.control_dependencies([du.assert_close(x, y)]):
117 with ops.control_dependencies([du.assert_close(y, z)]):
131 with ops.control_dependencies([du.assert_integer_form(x)]):
136 [du.assert_integer_form(y)]):
141 [du.assert_integer_form(z)]):
146 [du.assert_integer_form(w)]):
156 du.get_logits_and_probs(logits=None, probs=None)
159 du.get_logits_and_probs(logits=[0.1], probs=[0.1])
166 new_logits, new_p = du.get_logits_and_probs(
177 new_logits, new_p = du.get_logits_and_probs(
187 new_logits, new_p = du.get_logits_and_probs(
197 new_logits, new_p = du.get_logits_and_probs(
211 _, prob = du.get_logits_and_probs(
216 _, prob = du.get_logits_and_probs(
220 _, prob = du.get_logits_and_probs(
225 _, prob = du.get_logits_and_probs(
229 _, prob = du.get_logits_and_probs(
243 _, prob = du.get_logits_and_probs(
248 _, prob = du.get_logits_and_probs(
252 _, prob = du.get_logits_and_probs(
258 _, prob = du.get_logits_and_probs(
262 _, prob = du.get_logits_and_probs(
267 _, prob = du.get_logits_and_probs(
271 _, prob = du.get_logits_and_probs(
279 du.get_logits_and_probs(
285 _, prob = du.get_logits_and_probs(
293 du.get_logits_and_probs(
299 logit, _ = du.get_logits_and_probs(
311 checked_param = du.embed_check_categorical_event_shape(
317 checked_param = du.embed_check_categorical_event_shape(
325 checked_param = du.embed_check_categorical_event_shape(
331 checked_param = du.embed_check_categorical_event_shape(
339 du.embed_check_categorical_event_shape(param)
349 x_checked = du.embed_check_integer_casting_closed(
357 x_checked = du.embed_check_integer_casting_closed(
365 x_checked = du.embed_check_integer_casting_closed(
373 x_checked = du.embed_check_integer_casting_closed(
393 log_binom = du.log_combinations(n, counts)
405 log_binom = du.log_combinations(n, counts)
429 du.same_dynamic_shape(scalar, scalar1).eval({
436 du.same_dynamic_shape(vector, vector1).eval({
440 du.same_dynamic_shape(vector1, vector2).eval({
447 du.same_dynamic_shape(
452 du.same_dynamic_shape(
460 du.same_dynamic_shape(scalar, vector1).eval({
464 du.same_dynamic_shape(scalar1, vector1).eval({
469 du.same_dynamic_shape(scalar, multidimensional1).eval({
473 du.same_dynamic_shape(scalar1, multidimensional1).eval(
481 du.same_dynamic_shape(vector, vector1).eval({
485 du.same_dynamic_shape(vector1, vector2).eval({
490 du.same_dynamic_shape(vector, multidimensional1).eval({
494 du.same_dynamic_shape(vector1, multidimensional1).eval(
502 du.same_dynamic_shape(
507 du.same_dynamic_shape(
525 du.rotate_transpose(None, 1)
528 y = du.rotate_transpose(x, shift)
543 sess.run(du.rotate_transpose(x, shift),
556 du.pick_vector(
559 du.pick_vector(
562 du.pick_vector(
565 du.pick_vector(
574 rank = du.prefer_static_rank(x)
580 rank = du.prefer_static_rank(x)
586 rank = du.prefer_static_rank(x)
592 rank = du.prefer_static_rank(x)
598 rank = du.prefer_static_rank(x)
604 rank = du.prefer_static_rank(x)
614 shape = du.prefer_static_shape(x)
620 shape = du.prefer_static_shape(x)
626 shape = du.prefer_static_shape(x)
632 shape = du.prefer_static_shape(x)
638 shape = du.prefer_static_shape(x)
644 shape = du.prefer_static_shape(x)
654 value = du.prefer_static_value(x)
660 value = du.prefer_static_value(x)
666 value = du.prefer_static_value(x)
672 value = du.prefer_static_value(x)
679 value = du.prefer_static_value(x)
685 value = du.prefer_static_value(x)
732 actual = du.fill_triangular(x, **kwargs)
804 actual, actual_sgn = du.reduce_weighted_logsumexp(
826 actual, actual_sgn = du.reduce_weighted_logsumexp(
844 actual, actual_sgn = du.reduce_weighted_logsumexp(
862 du.reduce_weighted_logsumexp(x, w).eval())
867 du.reduce_weighted_logsumexp(x, w, axis=0).eval())
871 du.reduce_weighted_logsumexp(x, w, axis=1).eval())
875 du.reduce_weighted_logsumexp(x, w, axis=1, keep_dims=True).eval())
879 du.reduce_weighted_logsumexp(x, w, axis=[0, 1]).eval())
886 self.assertFalse(du.gen_new_seed(0, "salt") is None)
887 self.assertTrue(du.gen_new_seed(None, "salt") is None)
906 softplus_inverse = du
978 y = du.softplus_inverse(x)
988 y = du.softplus_inverse(x)
