| 42 | } |
| 43 | |
| 44 | TEST(BCastTest, Invalid) { |
| 45 | for (const bool use_optimization : {true, false}) { |
| 46 | EXPECT_EQ("invalid", BCast({5, 3, 2}, {3}, use_optimization)); |
| 47 | EXPECT_EQ("invalid", BCast({5, 3, 2}, {2, 2}, use_optimization)); |
| 48 | EXPECT_EQ("invalid", BCast({5, 3, 2}, {10, 1, 1}, use_optimization)); |
| 49 | EXPECT_EQ("invalid", |
| 50 | BCast({1, 2, 1, 2, 1, 2}, {2, 4, 2, 1, 2, 1}, use_optimization)); |
| 51 | } |
| 52 | } |
| 53 | |
| 54 | TEST(BCastTest, Basic_SameShape) { |
| 55 | // Effectively no broadcast needed. |