Lines Matching refs:num_elements
40 EXPECT_EQ(s.num_elements(), 1);
49 EXPECT_EQ(100, s.num_elements());
54 EXPECT_EQ(40, s.num_elements());
60 EXPECT_EQ(5, s.num_elements());
69 EXPECT_EQ(20000, s.num_elements());
78 EXPECT_EQ(30, s.num_elements());
91 ASSERT_EQ(210, s0.num_elements());
97 ASSERT_EQ(210, s0.num_elements());
104 ASSERT_EQ(42, s1.num_elements());
110 ASSERT_EQ(210, s2.num_elements());
117 ASSERT_EQ(42, s3.num_elements());
156 EXPECT_EQ(1000, s.num_elements());
158 EXPECT_EQ(0, s.num_elements());
160 EXPECT_EQ(1400, s.num_elements());
275 int64 num_elements() const { return num_elements_; }
348 int64 num_elements = 1;
351 num_elements *= d.size();
352 if (num_elements > kMaxElements) return false;
358 int64 num_elements = 1;
365 num_elements *= d.size();
366 if (num_elements > kMaxElements) {
524 EXPECT_EQ(s.num_elements(), sold.num_elements());
534 int64 ne = sold.num_elements();
537 int64 sz = SkewedSize(&gen, sold.num_elements());
549 int64 sz = SkewedSize(&gen, sold.num_elements());
555 int64 num_elements = 1;
557 int64 sz = SkewedSize(&gen, num_elements);
559 num_elements *= std::max<int64>(1, sz);
573 EXPECT_EQ(TensorShape({max}).num_elements(), max);
574 EXPECT_EQ(TensorShape({1, max}).num_elements(), max);
575 EXPECT_EQ(TensorShape({max, 1}).num_elements(), max);
576 EXPECT_EQ(TensorShape({one << 62}).num_elements(), one << 62);
577 EXPECT_EQ(TensorShape({one << 20, one << 41}).num_elements(), one << 61);
578 EXPECT_EQ(TensorShape({1000, 1000, 1000, 1000, 1000, 1000}).num_elements(),
