| 33 | } // namespace |
| 34 | |
| 35 | TEST(DescriptorPoolRegistryTest, TestBasic) { |
| 36 | EXPECT_EQ(DescriptorPoolRegistry::Global()->Get("NON-EXISTENT"), nullptr); |
| 37 | auto pool1 = DescriptorPoolRegistry::Global()->Get("TEST POOL 1"); |
| 38 | EXPECT_NE(pool1, nullptr); |
| 39 | auto pool2 = DescriptorPoolRegistry::Global()->Get("TEST POOL 2"); |
| 40 | EXPECT_NE(pool2, nullptr); |
| 41 | } |
| 42 | |
| 43 | } // namespace tensorflow |