| 24 | namespace { |
| 25 | |
| 26 | TEST(KernelDefBuilderTest, Basic) { |
| 27 | const KernelDef* def = KernelDefBuilder("A").Device(DEVICE_CPU).Build(); |
| 28 | KernelDef expected; |
| 29 | protobuf::TextFormat::ParseFromString("op: 'A' device_type: 'CPU'", |
| 30 | &expected); |
| 31 | EXPECT_EQ(def->DebugString(), expected.DebugString()); |
| 32 | delete def; |
| 33 | } |
| 34 | |
| 35 | TEST(KernelDefBuilderTest, TypeConstraint) { |
| 36 | const KernelDef* def = KernelDefBuilder("B") |
nothing calls this directly
no test coverage detected