Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:sample

61       x = dist.sample().eval()
89 self.assertAllEqual([3], dist.sample().eval().shape)
90 self.assertAllEqual([5, 3], dist.sample(5).eval().shape)
95 self.assertAllEqual([2, 3], dist.sample().eval().shape)
96 self.assertAllEqual([5, 2, 3], dist.sample(5).eval().shape)
101 self.assertAllEqual([4, 1, 3], dist.sample().eval().shape)
102 self.assertAllEqual([5, 4, 1, 3], dist.sample(5).eval().shape)
123 x = dist.sample().eval()
133 x = dist.sample().eval()
163 self.assertAllEqual([3], dist.sample().eval(feed_dict=feed_dict).shape)
165 dist.sample(5).eval(feed_dict=feed_dict).shape)
174 dist.log_prob(dist.sample())