| 62 | } |
| 63 | |
| 64 | OpSegment::CreateKernelFn GetFn(const NodeDef* ndef) { |
| 65 | return [this, ndef](OpKernel** kernel) { |
| 66 | Status s; |
| 67 | auto created = CreateOpKernel(DEVICE_CPU, &device_, cpu_allocator(), |
| 68 | *ndef, TF_GRAPH_DEF_VERSION, &s); |
| 69 | if (s.ok()) { |
| 70 | *kernel = created.release(); |
| 71 | } |
| 72 | return s; |
| 73 | }; |
| 74 | } |
| 75 | }; |
| 76 | |
| 77 | TEST_F(OpSegmentTest, Basic) { |
nothing calls this directly
no test coverage detected