Searched refs:get_ndims (Results 1 – 2 of 2) sorted by relevance
248 def get_ndims(self, x, name="get_ndims"): member in _DistributionShape279 ndims = self.get_ndims(x, name=name)
556 self.assertAllEqual(3, shaper.get_ndims(x).eval())562 self.assertAllEqual(3, shaper.get_ndims(x).eval())569 self.assertEqual(3, shaper.get_ndims(y).eval())583 self.assertEqual(2, sess.run(shaper.get_ndims(y), feed_dict=feed_dict))