diff options
Diffstat (limited to 'tensorflow/core/kernels/data/zip_dataset_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/data/zip_dataset_op.cc | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/data/zip_dataset_op.cc b/tensorflow/core/kernels/data/zip_dataset_op.cc index 00705236f9..63e9b99d4b 100644 --- a/tensorflow/core/kernels/data/zip_dataset_op.cc +++ b/tensorflow/core/kernels/data/zip_dataset_op.cc @@ -77,13 +77,14 @@ class ZipDatasetOp : public DatasetOpKernel { string DebugString() const override { return "ZipDatasetOp::Dataset"; } protected: - Status AsGraphDefInternal(OpKernelContext* ctx, DatasetGraphDefBuilder* b, + Status AsGraphDefInternal(SerializationContext* ctx, + DatasetGraphDefBuilder* b, Node** output) const override { std::vector<Node*> input_graph_nodes; input_graph_nodes.reserve(inputs_.size()); for (const auto& input : inputs_) { Node* input_node; - TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input, &input_node)); + TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &input_node)); input_graph_nodes.emplace_back(input_node); } TF_RETURN_IF_ERROR(b->AddDataset( @@ -142,7 +143,7 @@ class ZipDatasetOp : public DatasetOpKernel { writer->WriteScalar(full_name("input_impls_empty"), "")); } else { for (auto& input_impl : input_impls_) - TF_RETURN_IF_ERROR(SaveParent(writer, input_impl)); + TF_RETURN_IF_ERROR(SaveInput(writer, input_impl)); } return Status::OK(); } @@ -155,7 +156,7 @@ class ZipDatasetOp : public DatasetOpKernel { } else { DCHECK_EQ(input_impls_.size(), dataset()->inputs_.size()); for (auto& input_impl : input_impls_) - TF_RETURN_IF_ERROR(RestoreParent(ctx, reader, input_impl)); + TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl)); } return Status::OK(); } |