diff --git a/tensorflow/core/kernels/data/experimental/io_ops.cc b/tensorflow/core/kernels/data/experimental/io_ops.cc index 112a58a7e9a451..903088ff481390 100644 --- a/tensorflow/core/kernels/data/experimental/io_ops.cc +++ b/tensorflow/core/kernels/data/experimental/io_ops.cc @@ -28,6 +28,11 @@ namespace tensorflow { namespace data { namespace experimental { +/* static */ constexpr const int SaveDatasetOp::kFileFormatVersion; +/* static */ constexpr const char* const LoadDatasetOp::kCompression; +/* static */ constexpr const char* const LoadDatasetOp::kReaderFunc; +/* static */ constexpr const char* const LoadDatasetOp::kReaderFuncTarguments; + SaveDatasetOp::SaveDatasetOp(OpKernelConstruction* ctx) : HybridAsyncOpKernel(ctx, "tf_data_save_dataset") { OP_REQUIRES_OK(ctx, ctx->GetAttr(kCompression, &compression_)); diff --git a/tensorflow/core/kernels/data/experimental/snapshot_dataset_op.cc b/tensorflow/core/kernels/data/experimental/snapshot_dataset_op.cc index c029f2a50b815d..28935d92cd21b6 100644 --- a/tensorflow/core/kernels/data/experimental/snapshot_dataset_op.cc +++ b/tensorflow/core/kernels/data/experimental/snapshot_dataset_op.cc @@ -63,6 +63,15 @@ namespace tensorflow { namespace data { namespace experimental { +/* static */ constexpr const char* const SnapshotDatasetV2Op::kCompression; +/* static */ constexpr const char* const SnapshotDatasetV2Op::kReaderFunc; +/* static */ constexpr const char* const SnapshotDatasetV2Op::kShardFunc; +/* static */ constexpr const char* const + SnapshotDatasetV2Op::kReaderFuncTarguments; +/* static */ constexpr const char* const + SnapshotDatasetV2Op::kShardFuncTarguments; +/* static */ constexpr const int SnapshotDatasetV2Op::kFileFormatVersion; + // ==== Snapshot Implementation ==== /* The current snapshot on-disk layout is as follows: @@ -1122,9 +1131,7 @@ class SnapshotDatasetOp : public UnaryDatasetOpKernel { // Initialize at first and at that point we don't know which iterator // (Reader / Writer / Passthrough) we need to restore as this info is part // of the checkpoint. - Status Initialize(IteratorContext* ctx) override { - return Status::OK(); - } + Status Initialize(IteratorContext* ctx) override { return Status::OK(); } Status GetNextInternal(IteratorContext* ctx, std::vector* out_tensors,