From cf330efa207d635efecfb1b703ee65ccfbc3f98c Mon Sep 17 00:00:00 2001 From: Derek Murray Date: Fri, 2 Feb 2018 13:39:38 -0800 Subject: [PATCH] [tf.data] Move framework/dataset.h to framework/dataset_stateful_op_whitelist.h. This will make way for the move of kernels/data/dataset.h to framework/dataset.h, while preserving version control history, after which we might recombine the headers. PiperOrigin-RevId: 184327481 --- tensorflow/core/BUILD | 2 +- .../core/framework/{dataset.h => dataset_stateful_op_whitelist.h} | 6 +++--- tensorflow/core/kernels/data/dataset.h | 2 +- tensorflow/core/ops/lookup_ops.cc | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) rename tensorflow/core/framework/{dataset.h => dataset_stateful_op_whitelist.h} (93%) diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 1fcdfa7..68c82ea 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -432,7 +432,7 @@ tf_cuda_library( "framework/cancellation.h", "framework/common_shape_fns.h", "framework/control_flow.h", # TODO(josh11b): Make internal? - "framework/dataset.h", + "framework/dataset_stateful_op_whitelist.h", "framework/device_base.h", "framework/function.h", "framework/graph_def_util.h", diff --git a/tensorflow/core/framework/dataset.h b/tensorflow/core/framework/dataset_stateful_op_whitelist.h similarity index 93% rename from tensorflow/core/framework/dataset.h rename to tensorflow/core/framework/dataset_stateful_op_whitelist.h index f866183..3b48999 100644 --- a/tensorflow/core/framework/dataset.h +++ b/tensorflow/core/framework/dataset_stateful_op_whitelist.h @@ -13,8 +13,8 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ -#ifndef TENSORFLOW_FRAMEWORK_DATASET_H_ -#define TENSORFLOW_FRAMEWORK_DATASET_H_ +#ifndef TENSORFLOW_CORE_FRAMEWORK_DATASET_STATEFUL_OP_WHITELIST_H_ +#define TENSORFLOW_CORE_FRAMEWORK_DATASET_STATEFUL_OP_WHITELIST_H_ #include "tensorflow/core/lib/core/status.h" @@ -74,4 +74,4 @@ class WhitelistedStatefulOpRegistry { } // namespace tensorflow -#endif // TENSORFLOW_FRAMEWORK_DATASET_H_ +#endif // TENSORFLOW_CORE_FRAMEWORK_DATASET_STATEFUL_OP_WHITELIST_H_ diff --git a/tensorflow/core/kernels/data/dataset.h b/tensorflow/core/kernels/data/dataset.h index 08c3ca8..8238661 100644 --- a/tensorflow/core/kernels/data/dataset.h +++ b/tensorflow/core/kernels/data/dataset.h @@ -19,7 +19,7 @@ limitations under the License. #include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/framework/attr_value_util.h" -#include "tensorflow/core/framework/dataset.h" +#include "tensorflow/core/framework/dataset_stateful_op_whitelist.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/node_def.pb.h" diff --git a/tensorflow/core/ops/lookup_ops.cc b/tensorflow/core/ops/lookup_ops.cc index 50ea8ad..444aa8b 100644 --- a/tensorflow/core/ops/lookup_ops.cc +++ b/tensorflow/core/ops/lookup_ops.cc @@ -14,7 +14,7 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/core/framework/common_shape_fns.h" -#include "tensorflow/core/framework/dataset.h" +#include "tensorflow/core/framework/dataset_stateful_op_whitelist.h" #include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_def_builder.h" #include "tensorflow/core/framework/shape_inference.h" -- 2.7.4