diff options
author | Yong Tang <yong.tang.github@outlook.com> | 2018-07-02 07:41:42 -0700 |
---|---|---|
committer | Derek Murray <derek.murray@gmail.com> | 2018-07-02 07:41:42 -0700 |
commit | a7b7aa856f34bf2e44fbeb91d817742c61483618 (patch) | |
tree | c2fcf698127c4d330e7ba629ec4ae512efbd5b1a /tensorflow/core/platform/default/build_config.bzl | |
parent | 7e8927e7af0c51ac20a63bd4eab6ff83df1a39ae (diff) |
Add KinesisDataset support for tensorflow Dataset (#19712)
* Add KinesisDataset support for tensorflow Dataset
This fix is an attempt to add Kinesis support
for tensorflow's Dataset. Kinesis is provided by
AWS as a managed data streaming service. It is
similiar to Apache Kafka, often used in places
where maintaining a independent Kafka cluster on AWS
is not desirable or possible.
This fix adds the Kinesis support for tensorflow Dataset.
Similiar to the Kafka integration in tensorflow,
KinesisDataset outputs tf.string for records.
Test cases have also been added, which could be invoked manually.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Expose KinesisDataset in dataset_ops.cc
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Expose KinesisDataset in python wrapper
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add test cases for KinesisDataset
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Update AWS library include files
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add Bazel BUILD files
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Rename s3_crypto to aws_crypto
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Rename with_s3_support to with_aws_support
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Selectively add kinesis to tensorflow/contrib/BUILD
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Set different partition key and pylint fix.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add missing modules in cmake's python_modules.txt
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Address review feedback
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/core/platform/default/build_config.bzl')
-rw-r--r-- | tensorflow/core/platform/default/build_config.bzl | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/core/platform/default/build_config.bzl b/tensorflow/core/platform/default/build_config.bzl index 66ccd81e41..28891320c4 100644 --- a/tensorflow/core/platform/default/build_config.bzl +++ b/tensorflow/core/platform/default/build_config.bzl @@ -620,10 +620,10 @@ def tf_additional_core_deps(): ], "//conditions:default": [], }) + select({ - "//tensorflow:with_s3_support_windows_override": [], - "//tensorflow:with_s3_support_android_override": [], - "//tensorflow:with_s3_support_ios_override": [], - "//tensorflow:with_s3_support": [ + "//tensorflow:with_aws_support_windows_override": [], + "//tensorflow:with_aws_support_android_override": [], + "//tensorflow:with_aws_support_ios_override": [], + "//tensorflow:with_aws_support": [ "//tensorflow/core/platform/s3:s3_file_system", ], "//conditions:default": [], |