aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/cmake/tf_core_ops.cmake
diff options
context:
space:
mode:
authorGravatar Derek Murray <mrry@google.com>2017-11-07 14:25:00 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-11-07 14:29:06 -0800
commitdacce477d4831540dda1e68d6d0f8f017aab0020 (patch)
tree2086c2df34342d518dd3644d42ca74ae81f46606 /tensorflow/contrib/cmake/tf_core_ops.cmake
parentd0de8738e3401bbc5fd142846b4fc124951e5e07 (diff)
[tf.data] Move contrib op registrations to tensorflow/contrib/data/ops.
This change moves the contrib-level Dataset ops out of the core op libraries (where they were placed for short-term technical reasons) into a more appropriate location in contrib. This enables us to modify the signatures of these ops without being subject to the core backwards compatibility requirements. (We currently must modify the backwards compatibility ledger each time we make an allowed change to a contrib op.) For now, the kernel implementations remain in the core library, because they depend on headers that are not part of the public API. The change also removes some code testing contrib features in the core kernel_tests, since it relies on the contrib op registrations and was already adequately tested in the contrib tests. PiperOrigin-RevId: 174912490
Diffstat (limited to 'tensorflow/contrib/cmake/tf_core_ops.cmake')
-rw-r--r--tensorflow/contrib/cmake/tf_core_ops.cmake1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/tf_core_ops.cmake b/tensorflow/contrib/cmake/tf_core_ops.cmake
index 4a61ed7a35..03c168795c 100644
--- a/tensorflow/contrib/cmake/tf_core_ops.cmake
+++ b/tensorflow/contrib/cmake/tf_core_ops.cmake
@@ -81,6 +81,7 @@ GENERATE_CONTRIB_OP_LIBRARY(boosted_trees_prediction "${tensorflow_source_dir}/t
GENERATE_CONTRIB_OP_LIBRARY(boosted_trees_quantiles "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/ops/quantile_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(boosted_trees_stats_accumulator "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/ops/stats_accumulator_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(cudnn_rnn "${tensorflow_source_dir}/tensorflow/contrib/cudnn_rnn/ops/cudnn_rnn_ops.cc")
+GENERATE_CONTRIB_OP_LIBRARY(data_dataset "${tensorflow_source_dir}/tensorflow/contrib/data/ops/dataset_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(data_prefetching "${tensorflow_source_dir}/tensorflow/contrib/data/ops/prefetching_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(factorization_clustering "${tensorflow_source_dir}/tensorflow/contrib/factorization/ops/clustering_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(factorization_factorization "${tensorflow_source_dir}/tensorflow/contrib/factorization/ops/factorization_ops.cc")