diff options
author | Derek Murray <mrry@google.com> | 2018-09-05 18:16:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-05 18:20:28 -0700 |
commit | ad5c0c4d091c93ef65e91c55cb4df065d0c7a989 (patch) | |
tree | 7a2fb348f4fab36e3b65963a7613c3ac8f8f21a4 /tensorflow/contrib/bigtable | |
parent | 7ec8114697a78271277c1b81707f53057d047901 (diff) |
[tf.data] Move all C++ code inside the `tensorflow::data` namespace.
PiperOrigin-RevId: 211733735
Diffstat (limited to 'tensorflow/contrib/bigtable')
8 files changed, 22 insertions, 0 deletions
diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc b/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc index a25a641cdb..6138d79126 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc @@ -172,6 +172,11 @@ class BigtableTableOp : public OpKernel { REGISTER_KERNEL_BUILDER(Name("BigtableTable").Device(DEVICE_CPU), BigtableTableOp); +} // namespace + +namespace data { +namespace { + class ToBigtableOp : public AsyncOpKernel { public: explicit ToBigtableOp(OpKernelConstruction* ctx) @@ -354,5 +359,6 @@ REGISTER_KERNEL_BUILDER(Name("DatasetToBigtable").Device(DEVICE_CPU), ToBigtableOp); } // namespace +} // namespace data } // namespace tensorflow diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_lib.h b/tensorflow/contrib/bigtable/kernels/bigtable_lib.h index a2a5df1037..4652021fec 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_lib.h +++ b/tensorflow/contrib/bigtable/kernels/bigtable_lib.h @@ -79,6 +79,8 @@ class BigtableTableResource : public ResourceBase { ::google::cloud::bigtable::noex::Table table_; }; +namespace data { + // BigtableReaderDatasetIterator is an abstract class for iterators from // datasets that are "readers" (source datasets, not transformation datasets) // that read from Bigtable. @@ -138,6 +140,8 @@ class BigtableReaderDatasetIterator : public DatasetIterator<Dataset> { ::google::cloud::bigtable::RowReader::iterator iterator_ GUARDED_BY(mu_); }; +} // namespace data + } // namespace tensorflow #endif // TENSORFLOW_CONTRIB_BIGTABLE_KERNELS_BIGTABLE_LIB_H_ diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_lookup_dataset_op.cc b/tensorflow/contrib/bigtable/kernels/bigtable_lookup_dataset_op.cc index bd32672aa9..11f530e82a 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_lookup_dataset_op.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_lookup_dataset_op.cc @@ -17,6 +17,7 @@ limitations under the License. #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { +namespace data { namespace { class BigtableLookupDatasetOp : public UnaryDatasetOpKernel { @@ -226,4 +227,5 @@ REGISTER_KERNEL_BUILDER(Name("BigtableLookupDataset").Device(DEVICE_CPU), BigtableLookupDatasetOp); } // namespace +} // namespace data } // namespace tensorflow diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_prefix_key_dataset_op.cc b/tensorflow/contrib/bigtable/kernels/bigtable_prefix_key_dataset_op.cc index a803fdcb49..5cab729d9c 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_prefix_key_dataset_op.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_prefix_key_dataset_op.cc @@ -17,6 +17,7 @@ limitations under the License. #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { +namespace data { namespace { class BigtablePrefixKeyDatasetOp : public DatasetOpKernel { @@ -111,4 +112,5 @@ REGISTER_KERNEL_BUILDER(Name("BigtablePrefixKeyDataset").Device(DEVICE_CPU), BigtablePrefixKeyDatasetOp); } // namespace +} // namespace data } // namespace tensorflow diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_range_key_dataset_op.cc b/tensorflow/contrib/bigtable/kernels/bigtable_range_key_dataset_op.cc index 5cd0371c79..4dc4647bd2 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_range_key_dataset_op.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_range_key_dataset_op.cc @@ -17,6 +17,7 @@ limitations under the License. #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { +namespace data { namespace { class BigtableRangeKeyDatasetOp : public DatasetOpKernel { @@ -117,4 +118,5 @@ class BigtableRangeKeyDatasetOp : public DatasetOpKernel { REGISTER_KERNEL_BUILDER(Name("BigtableRangeKeyDataset").Device(DEVICE_CPU), BigtableRangeKeyDatasetOp); } // namespace +} // namespace data } // namespace tensorflow diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_sample_key_pairs_dataset_op.cc b/tensorflow/contrib/bigtable/kernels/bigtable_sample_key_pairs_dataset_op.cc index 6928d9423c..736775bdac 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_sample_key_pairs_dataset_op.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_sample_key_pairs_dataset_op.cc @@ -18,6 +18,7 @@ limitations under the License. #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { +namespace data { namespace { class BigtableSampleKeyPairsDatasetOp : public DatasetOpKernel { @@ -205,4 +206,5 @@ REGISTER_KERNEL_BUILDER( BigtableSampleKeyPairsDatasetOp); } // namespace +} // namespace data } // namespace tensorflow diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_sample_keys_dataset_op.cc b/tensorflow/contrib/bigtable/kernels/bigtable_sample_keys_dataset_op.cc index a759fb5063..208b7b3e08 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_sample_keys_dataset_op.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_sample_keys_dataset_op.cc @@ -17,6 +17,7 @@ limitations under the License. #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { +namespace data { namespace { class BigtableSampleKeysDatasetOp : public DatasetOpKernel { @@ -118,4 +119,5 @@ REGISTER_KERNEL_BUILDER(Name("BigtableSampleKeysDataset").Device(DEVICE_CPU), BigtableSampleKeysDatasetOp); } // namespace +} // namespace data } // namespace tensorflow diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_scan_dataset_op.cc b/tensorflow/contrib/bigtable/kernels/bigtable_scan_dataset_op.cc index 78a920b077..9407855fe8 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_scan_dataset_op.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_scan_dataset_op.cc @@ -17,6 +17,7 @@ limitations under the License. #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { +namespace data { namespace { class BigtableScanDatasetOp : public DatasetOpKernel { @@ -224,4 +225,5 @@ REGISTER_KERNEL_BUILDER(Name("BigtableScanDataset").Device(DEVICE_CPU), BigtableScanDatasetOp); } // namespace +} // namespace data } // namespace tensorflow |