mirror of
https://github.com/microsoft/vcpkg.git
synced 2024-12-23 09:40:35 +08:00
65 lines
2.3 KiB
Diff
65 lines
2.3 KiB
Diff
|
diff --git a/third_party/aws/aws-checksums.bazel b/third_party/aws/aws-checksums.bazel
|
||
|
|
--- a/third_party/aws/aws-checksums.bazel
|
||
|
|
+++ b/third_party/aws/aws-checksums.bazel
|
||
|
|
@@ -7,6 +7,13 @@
|
||
|
|
|
||
|
|
exports_files(["LICENSE"])
|
||
|
|
|
||
|
|
+load("@bazel_skylib//lib:selects.bzl", "selects")
|
||
|
|
+
|
||
|
|
+selects.config_setting_group(
|
||
|
|
+ name = "is_linux_debug",
|
||
|
|
+ match_all = ["@org_tensorflow//tensorflow:linux_x86_64", "@org_tensorflow//tensorflow:debug"],
|
||
|
|
+)
|
||
|
|
+
|
||
|
|
cc_library(
|
||
|
|
name = "aws-checksums",
|
||
|
|
srcs = select({
|
||
|
|
@@ -27,4 +34,10 @@
|
||
|
|
deps = [
|
||
|
|
"@aws-c-common",
|
||
|
|
],
|
||
|
|
+ defines = select({
|
||
|
|
+ ":is_linux_debug": [
|
||
|
|
+ "DEBUG_BUILD"
|
||
|
|
+ ],
|
||
|
|
+ "//conditions:default": [],
|
||
|
|
+ }),
|
||
|
|
)
|
||
|
|
|
||
|
|
diff --git a/tensorflow/core/kernels/data/experimental/io_ops.cc b/tensorflow/core/kernels/data/experimental/io_ops.cc
|
||
|
|
--- a/tensorflow/core/kernels/data/experimental/io_ops.cc
|
||
|
|
+++ b/tensorflow/core/kernels/data/experimental/io_ops.cc
|
||
|
|
@@ -28,6 +28,12 @@
|
||
|
|
namespace data {
|
||
|
|
namespace experimental {
|
||
|
|
|
||
|
|
+/*static*/ constexpr const char* const LoadDatasetOp::kCompression;
|
||
|
|
+/*static*/ constexpr const char* const LoadDatasetOp::kReaderFunc;
|
||
|
|
+/*static*/ constexpr const char* const LoadDatasetOp::kReaderFuncTarguments;
|
||
|
|
+
|
||
|
|
+/*static*/ constexpr const int SaveDatasetOp::kFileFormatVersion;
|
||
|
|
+
|
||
|
|
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
|
||
|
|
--- a/tensorflow/core/kernels/data/experimental/snapshot_dataset_op.cc
|
||
|
|
+++ b/tensorflow/core/kernels/data/experimental/snapshot_dataset_op.cc
|
||
|
|
@@ -63,6 +63,14 @@
|
||
|
|
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:
|