Merge pull request #22210 from dmitrievanthony:apache-ignite-dataset

PiperOrigin-RevId: 215258743

1 files changed, 10 insertions, 0 deletions

diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD
index 98dff965a9..fbe0573d5d 100644
--- a/tensorflow/contrib/BUILD
+++ b/tensorflow/contrib/BUILD
@@ -123,6 +123,11 @@ py_library(
             "//tensorflow/contrib/tensorrt:init_py",
             "//tensorflow/contrib/ffmpeg:ffmpeg_ops_py",
         ],
+    }) + select({
+        "//tensorflow:with_ignite_support": [
+            "//tensorflow/contrib/ignite",
+        ],
+        "//conditions:default": [],
     }),
 )
 
@@ -184,5 +189,10 @@ cc_library(
             "//tensorflow/contrib/kinesis:dataset_ops_op_lib",
             "//tensorflow/contrib/tensorrt:trt_engine_op_op_lib",
         ],
+    }) + select({
+        "//tensorflow:with_ignite_support": [
+            "//tensorflow/contrib/ignite:dataset_ops_op_lib",
+        ],
+        "//conditions:default": [],
     }),
 )