diff options
author | 2017-11-07 14:25:00 -0800 | |
---|---|---|
committer | 2017-11-07 14:29:06 -0800 | |
commit | dacce477d4831540dda1e68d6d0f8f017aab0020 (patch) | |
tree | 2086c2df34342d518dd3644d42ca74ae81f46606 /tensorflow/contrib/cmake/tf_core_ops.cmake | |
parent | d0de8738e3401bbc5fd142846b4fc124951e5e07 (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.cmake | 1 |
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") |