| 373 | } // anonymous namespace |
| 374 | |
| 375 | Status AsGraphDef(OpKernelContext* ctx, const DatasetBase* dataset, |
| 376 | SerializationContext&& serialization_ctx, |
| 377 | GraphDef* graph_def) { |
| 378 | if (serialization_ctx.check_external_state()) { |
| 379 | TF_RETURN_IF_ERROR(dataset->CheckExternalState()); |
| 380 | } |
| 381 | GraphDefBuilder b; |
| 382 | DatasetBase::DatasetGraphDefBuilder db(&b); |
| 383 | Node* output_node = nullptr; |
| 384 | TF_RETURN_IF_ERROR( |
| 385 | db.AddInputDataset(&serialization_ctx, dataset, &output_node)); |
| 386 | // Insert a purely symbolic _Retval node to indicate to consumers which node |
| 387 | // represents `dataset`. |
| 388 | ops::UnaryOp("_Retval", output_node, |
| 389 | b.opts() |
| 390 | .WithName("dataset") |
| 391 | .WithAttr("T", DT_VARIANT) |
| 392 | .WithAttr("index", 0)); |
| 393 | TF_RETURN_IF_ERROR(b.ToGraphDef(graph_def)); |
| 394 | return Status::OK(); |
| 395 | } |
| 396 | |
| 397 | Status ConnectCancellationManagers(CancellationManager* parent, |
| 398 | CancellationManager* child, |
no test coverage detected