Lines Matching refs:math_ops
15 """Tests for math_ops.bincount."""
25 from tensorflow.python.ops import math_ops
34 math_ops.bincount([], minlength=5).eval(), [0, 0, 0, 0, 0])
35 self.assertAllEqual(math_ops.bincount([], minlength=1).eval(), [0])
36 self.assertAllEqual(math_ops.bincount([], minlength=0).eval(), [])
38 math_ops.bincount([], minlength=0, dtype=np.float32).eval().dtype,
41 math_ops.bincount([], minlength=3, dtype=np.float64).eval().dtype,
47 math_ops.bincount([1, 1, 1, 2, 2, 3]).eval(), [0, 3, 2, 1])
49 self.assertAllEqual(math_ops.bincount(arr).eval(), [0, 5, 4, 3, 2, 1])
51 self.assertAllEqual(math_ops.bincount(arr).eval(), [6, 5, 4, 3, 2, 1])
53 self.assertAllEqual(math_ops.bincount([]).eval(), [])
54 self.assertAllEqual(math_ops.bincount([0, 0, 0]).eval(), [3])
55 self.assertAllEqual(math_ops.bincount([5]).eval(), [0, 0, 0, 0, 0, 1])
57 math_ops.bincount(np.arange(10000)).eval(), np.ones(10000))
61 self.assertAllEqual(math_ops.bincount([5], maxlength=3).eval(), [0, 0, 0])
62 self.assertAllEqual(math_ops.bincount([1], maxlength=3).eval(), [0, 1])
63 self.assertAllEqual(math_ops.bincount([], maxlength=3).eval(), [])
76 math_ops.bincount(arr, weights).eval(), np.bincount(arr, weights))
86 math_ops.bincount(arr, None).eval(), np.bincount(arr, weights))
91 math_ops.bincount(np.arange(1000), np.zeros(1000)).eval(),
98 math_ops.bincount([1, 2, 3, -1, 6, 8]).eval()
