| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 154 class Iterator : public DatasetIterator<Dataset<T>> {
157 : DatasetIterator<Dataset<T>>(params) {} in Iterator()
160 return DatasetIterator<Dataset<T>>::dataset()->input_->MakeIterator( in Initialize()
161 ctx, this, DatasetIterator<Dataset<T>>::prefix(), &input_impl_); in Initialize()
172 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in GetNextInternal()
174 DatasetIterator<Dataset<T>>::dataset()->row_shape_; in GetNextInternal()
193 i < DatasetIterator<Dataset<T>>::dataset()->batch_size_ && in GetNextInternal()
243 DatasetIterator<Dataset<T>>::dataset()->input_->output_dtypes()[0], in GetNextInternal()
295 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in CreateNode()
|
| D | stats_dataset_ops.cc | 104 class Iterator : public DatasetIterator<Dataset> {
107 : DatasetIterator<Dataset>(params) {} in Iterator()
221 class Iterator : public DatasetIterator<Dataset> {
224 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | threadpool_dataset_op.cc | 200 class Iterator : public DatasetIterator<Dataset> {
203 : DatasetIterator<Dataset>(params) {} in Iterator()
332 class Iterator : public DatasetIterator<Dataset> {
335 : DatasetIterator<Dataset>(params) {} in Iterator()
456 class Iterator : public DatasetIterator<Dataset> {
459 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | non_serializable_dataset_op.cc | 93 class Iterator : public DatasetIterator<Dataset> {
96 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | ignore_errors_dataset_op.cc | 94 class Iterator : public DatasetIterator<Dataset> {
97 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | sleep_dataset_op.cc | 101 class Iterator : public DatasetIterator<Dataset> {
104 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | random_dataset_op.cc | 86 class Iterator : public DatasetIterator<Dataset> {
89 : DatasetIterator<Dataset>(params), in Iterator()
|
| D | assert_next_dataset_op.cc | 90 class Iterator : public DatasetIterator<Dataset> {
93 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | assert_cardinality_dataset_op.cc | 94 class Iterator : public DatasetIterator<Dataset> {
97 : DatasetIterator<Dataset>(params), num_elements_(0) {} in Iterator()
|
| D | take_while_dataset_op.cc | 123 class Iterator : public DatasetIterator<Dataset> {
126 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | snapshot_dataset_op.cc | 162 class SnapshotDatasetV2Op::Dataset::Iterator : public DatasetIterator<Dataset> {
199 : public DatasetIterator<Dataset> {
233 : public DatasetIterator<Dataset> {
286 : public DatasetIterator<Dataset> {
429 : DatasetIterator<Dataset>(params), in Iterator()
549 : DatasetIterator<Dataset>(params), start_index_(start_index) {} in Reader()
628 : DatasetIterator<Dataset>(params), in Writer()
811 : DatasetIterator<Dataset>(params) {} in Passthrough()
1201 class Iterator : public DatasetIterator<Dataset> {
1204 : DatasetIterator<Dataset>(params) { in Iterator()
[all …]
|
| D | sql_dataset_op.cc | 130 class Iterator : public DatasetIterator<Dataset> {
133 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | unique_dataset_op.cc | 77 class Iterator : public DatasetIterator<Dataset> {
80 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | rebatch_dataset_op.cc | 112 class Iterator : public DatasetIterator<Dataset> {
115 : DatasetIterator<Dataset>(params) {} in Iterator()
369 class Iterator : public DatasetIterator<Dataset> {
372 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | choose_fastest_branch_dataset_op.cc | 84 class WrapperIterator : public DatasetIterator<WrapperDataset> {
87 : DatasetIterator<WrapperDataset>(params), error_(error) {} in WrapperIterator()
337 class ChooseFastestIterator : public DatasetIterator<Dataset> {
340 : DatasetIterator<Dataset>(params), in ChooseFastestIterator()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | repeat_dataset_op.cc | 114 class EmptyIterator : public DatasetIterator<Dataset> {
117 : DatasetIterator<Dataset>(params) {} in EmptyIterator()
142 class FiniteIterator : public DatasetIterator<Dataset> {
145 : DatasetIterator<Dataset>(params), i_(0) {} in FiniteIterator()
214 class ForeverIterator : public DatasetIterator<Dataset> {
217 : DatasetIterator<Dataset>(params), in ForeverIterator()
|
| D | skip_dataset_op.cc | 100 class EmptyIterator : public DatasetIterator<Dataset> {
103 : DatasetIterator<Dataset>(params) {} in EmptyIterator()
129 class FiniteIterator : public DatasetIterator<Dataset> {
132 : DatasetIterator<Dataset>(params), i_(0) {} in FiniteIterator()
|
| D | take_dataset_op.cc | 87 class TakeDataset::EmptyIterator : public DatasetIterator<TakeDataset> {
90 : DatasetIterator<TakeDataset>(params) {} in EmptyIterator()
115 class TakeDataset::FiniteIterator : public DatasetIterator<TakeDataset> {
118 : DatasetIterator<TakeDataset>(params), i_(0) {} in FiniteIterator()
|
| D | cache_dataset_ops.cc | 138 class FileIterator : public DatasetIterator<FileDatasetBase> {
141 : DatasetIterator<FileDatasetBase>(params) { in FileIterator()
219 class FileWriterIterator : public DatasetIterator<FileDatasetBase> {
222 : DatasetIterator<FileDatasetBase>(params), in FileWriterIterator()
489 class FileReaderIterator : public DatasetIterator<FileDatasetBase> {
492 : DatasetIterator<FileDatasetBase>(params), in FileReaderIterator()
709 class MemoryIterator : public DatasetIterator<MemoryDatasetBase> {
712 : DatasetIterator<MemoryDatasetBase>(params), cache_(cache) {} in MemoryIterator()
760 class MemoryWriterIterator : public DatasetIterator<MemoryDatasetBase> {
763 : DatasetIterator<MemoryDatasetBase>(params), cache_(cache) {} in MemoryWriterIterator()
[all …]
|
| D | window_dataset.cc | 86 class Iterator : public DatasetIterator<WindowDataset> {
89 : DatasetIterator<WindowDataset>(params) {} in Iterator()
|
| D | tensor_dataset_op.cc | 98 class Iterator : public DatasetIterator<Dataset> {
101 : DatasetIterator<Dataset>(params), produced_(false) {} in Iterator()
|
| D | zip_dataset_op.cc | 121 class Iterator : public DatasetIterator<Dataset> {
124 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | tensor_slice_dataset_op.cc | 111 class Iterator : public DatasetIterator<Dataset> {
114 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_side_effects.h | 56 struct DatasetIterator : ::mlir::SideEffects::Resource::Base<DatasetIterator> { struct
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 401 class DatasetIterator : public InitializableLookupTable::InitTableIterator { class
403 explicit DatasetIterator(data::DatasetBase* dataset) : dataset_(dataset) {} in DatasetIterator() function in tensorflow::lookup::DatasetIterator
405 ~DatasetIterator() override {} in ~DatasetIterator()
490 DatasetIterator iter(dataset); in InitializeTableFromDataset()
|