Home
last modified time | relevance | path

Searched refs:DatasetIterator (Results 1 – 25 of 73) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc146 class Iterator : public DatasetIterator<Dataset<T>> {
149 : DatasetIterator<Dataset<T>>(params) {} in Iterator()
152 return DatasetIterator<Dataset<T>>::dataset()->input_->MakeIterator( in Initialize()
153 ctx, DatasetIterator<Dataset<T>>::prefix(), &input_impl_); in Initialize()
164 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in GetNextInternal()
166 DatasetIterator<Dataset<T>>::dataset()->row_shape_; in GetNextInternal()
185 i < DatasetIterator<Dataset<T>>::dataset()->batch_size_ && in GetNextInternal()
235 DatasetIterator<Dataset<T>>::dataset()->input_->output_dtypes()[0], in GetNextInternal()
287 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in CreateNode()
Dthreadpool_dataset_op.cc189 class Iterator : public DatasetIterator<Dataset> {
192 : DatasetIterator<Dataset>(params) {} in Iterator()
297 class Iterator : public DatasetIterator<Dataset> {
300 : DatasetIterator<Dataset>(params) {} in Iterator()
396 class Iterator : public DatasetIterator<Dataset> {
399 : DatasetIterator<Dataset>(params) {} in Iterator()
Dstats_dataset_ops.cc93 class Iterator : public DatasetIterator<Dataset> {
96 : DatasetIterator<Dataset>(params) {} in Iterator()
203 class Iterator : public DatasetIterator<Dataset> {
206 : DatasetIterator<Dataset>(params) {} in Iterator()
Dnon_serializable_dataset_op.cc81 class Iterator : public DatasetIterator<Dataset> {
84 : DatasetIterator<Dataset>(params) {} in Iterator()
Dsleep_dataset_op.cc92 class Iterator : public DatasetIterator<Dataset> {
95 : DatasetIterator<Dataset>(params) {} in Iterator()
Dignore_errors_dataset_op.cc76 class Iterator : public DatasetIterator<Dataset> {
79 : DatasetIterator<Dataset>(params) {} in Iterator()
Dassert_next_dataset_op.cc95 class Iterator : public DatasetIterator<Dataset> {
98 : DatasetIterator<Dataset>(params) {} in Iterator()
Drandom_dataset_op.cc94 class Iterator : public DatasetIterator<Dataset> {
97 : DatasetIterator<Dataset>(params), in Iterator()
/external/tensorflow/tensorflow/core/kernels/data/
Drepeat_dataset_op.cc105 class EmptyIterator : public DatasetIterator<Dataset> {
108 : DatasetIterator<Dataset>(params) {} in EmptyIterator()
132 class FiniteIterator : public DatasetIterator<Dataset> {
135 : DatasetIterator<Dataset>(params), i_(0) {} in FiniteIterator()
201 class ForeverIterator : public DatasetIterator<Dataset> {
204 : DatasetIterator<Dataset>(params), in ForeverIterator()
Dtake_dataset_op.cc42 class TakeDataset::EmptyIterator : public DatasetIterator<TakeDataset> {
45 : DatasetIterator<TakeDataset>(params) {} in EmptyIterator()
69 class TakeDataset::FiniteIterator : public DatasetIterator<TakeDataset> {
72 : DatasetIterator<TakeDataset>(params), i_(0) {} in FiniteIterator()
Dskip_dataset_op.cc92 class EmptyIterator : public DatasetIterator<Dataset> {
95 : DatasetIterator<Dataset>(params) {} in EmptyIterator()
120 class FiniteIterator : public DatasetIterator<Dataset> {
123 : DatasetIterator<Dataset>(params), i_(0) {} in FiniteIterator()
Dcache_dataset_ops.cc112 class FileIterator : public DatasetIterator<FileDataset> {
115 : DatasetIterator<FileDataset>(params) { in FileIterator()
195 class FileWriterIterator : public DatasetIterator<FileDataset> {
198 : DatasetIterator<FileDataset>(params), in FileWriterIterator()
439 class FileReaderIterator : public DatasetIterator<FileDataset> {
442 : DatasetIterator<FileDataset>(params), in FileReaderIterator()
688 class MemoryIterator : public DatasetIterator<MemoryDataset> {
691 : DatasetIterator<MemoryDataset>(params) {} in MemoryIterator()
805 class MemoryWriterIterator : public DatasetIterator<MemoryDataset> {
808 : DatasetIterator<MemoryDataset>(params), cache_(cache) { in MemoryWriterIterator()
[all …]
Dwindow_dataset.cc66 class Iterator : public DatasetIterator<WindowDataset> {
69 : DatasetIterator<WindowDataset>(params) {} in Iterator()
Dreader_dataset_ops.cc131 class Iterator : public DatasetIterator<Dataset> {
134 : DatasetIterator<Dataset>(params) {} in Iterator()
406 class UncompressedIterator : public DatasetIterator<Dataset> {
409 : DatasetIterator<Dataset>(params) {} in UncompressedIterator()
529 class CompressedIterator : public DatasetIterator<Dataset> {
532 : DatasetIterator<Dataset>(params) {} in CompressedIterator()
822 class Iterator : public DatasetIterator<Dataset> {
825 : DatasetIterator<Dataset>(params) {} in Iterator()
Dfilter_by_component_dataset_op.cc101 class Iterator : public DatasetIterator<Dataset> {
104 : DatasetIterator<Dataset>(params) {} in Iterator()
Dtensor_dataset_op.cc97 class Iterator : public DatasetIterator<Dataset> {
100 : DatasetIterator<Dataset>(params), produced_(false) {} in Iterator()
Drange_dataset_op.cc99 class Iterator : public DatasetIterator<Dataset> {
102 : DatasetIterator<Dataset>(params) { in Iterator()
Dshard_dataset_op.cc111 class Iterator : public DatasetIterator<Dataset> {
114 : DatasetIterator<Dataset>(params), next_index_(0) {} in Iterator()
Dzip_dataset_op.cc111 class Iterator : public DatasetIterator<Dataset> {
114 : DatasetIterator<Dataset>(params) {} in Iterator()
Dtensor_slice_dataset_op.cc122 class Iterator : public DatasetIterator<Dataset> {
125 : DatasetIterator<Dataset>(params), in Iterator()
Dgenerator_dataset_op.cc68 class Iterator : public DatasetIterator<Dataset> {
71 : DatasetIterator<Dataset>(params) {} in Iterator()
Dmodel_dataset_op.cc89 class Iterator : public DatasetIterator<Dataset> {
92 : DatasetIterator<Dataset>(params) { in Iterator()
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_lib.h89 class BigtableReaderDatasetIterator : public DatasetIterator<Dataset> {
92 const typename DatasetIterator<Dataset>::Params& params) in BigtableReaderDatasetIterator()
93 : DatasetIterator<Dataset>(params) {} in BigtableReaderDatasetIterator()
Dbigtable_sample_keys_dataset_op.cc77 class Iterator : public DatasetIterator<Dataset> {
80 : DatasetIterator<Dataset>(params) {} in Iterator()
Dbigtable_sample_key_pairs_dataset_op.cc113 class Iterator : public DatasetIterator<Dataset> {
116 : DatasetIterator<Dataset>(params) {} in Iterator()

123